yunxi1

I train Yolo v7 on an RTX 3090. When batch=12 , out of memory will occur when epoch=3. However, the memory usage is 22/24.

I change my cuda to 11.0, then set environment variable export CUDA_HOME=/usr/local/cuda new error appeared： nvcc fatal : Unsupported gpu architecture 'compute_86' error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1...

thank you, it works

确实，我也觉得代码有问题，应该把residual相关代码删掉才符合论文的公式和图解

我两种情况都跑了一遍，不乘residual效果非常差，乘了会涨点一点点，所以也许是论文表述有问题

I have tried ways: https://github.com/thomasbrandon/mish-cuda and https://github.com/JunnYu/mish-cuda they all failed and I change my mind to use mish function in your code: class Mish(nn.Module): # https://github.com/digantamisra98/Mish def forward(self, x): return...

I will try mish_cuda in cuda11.0 again

cuda11.0 failed too this is error ,have you ever encountered this problem? My environment variable is right. :/usr/local/cuda/bin/nvcc -Iexternal -I/home/shen/software/anaconda3/envs/nvidia-tensorflow-onnx/lib/python3.6/site-packages/torch/include -I/home/shen/software/anaconda3/envs/nvidia-tensorflow-onnx/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/home/shen/software/anaconda3/envs/nvidia-tensorflow-onnx/lib/python3.6/site-packages/torch/include/TH -I/home/shen/software/anaconda3/envs/nvidia-tensorflow-onnx/lib/python3.6/site-packages/torch/include/THC -I:/usr/local/cuda/include -I/home/shen/software/anaconda3/envs/nvidia-tensorflow-onnx/include/python3.6m -c csrc/mish_kernel.cu -o build/temp.linux-x86_64-3.6/csrc/mish_kernel.o -D__CUDA_NO_HALF_OPERATORS__...

command: python setup.py install ways: https://github.com/thomasbrandon/mish-cuda or https://github.com/JunnYu/mish-cuda new error: nvcc fatal : Unsupported gpu architecture 'compute_86' error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

yes, add environment variable：export TORCH_CUDA_ARCH_LIST="7.5" downgrade version it works!