-
Notifications
You must be signed in to change notification settings - Fork 573
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Can't use mish-cuda in cuda 11.1 #119
Comments
i am not sure about cuda 11.1, but i can build mish_cuda with cuda 11.0. |
I have tried ways: https://github.com/thomasbrandon/mish-cuda and https://github.com/JunnYu/mish-cuda File "train.py", line 122, in train |
I will try mish_cuda in cuda11.0 again |
cuda11.0 failed too :/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__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' --expt-extended-lambda -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=sm_86 -std=c++14
|
what is the command you used to build mish_cuda? |
command: python setup.py install |
well, it seems compute_86 only supported by cuda>=11.1. |
yes, add environment variable:export TORCH_CUDA_ARCH_LIST="7.5" downgrade version |
@yunxi1 |
@hoangduyloc make sure that pytorch1.7+cuda11.0 installed correctly,then add environment variable:export TORCH_CUDA_ARCH_LIST="8.0" because pytorch cant't work in compute_86(3090),but it could work in compute_80,good luck! |
|
Hi, I have tried your method, but it could not work. unable to execute '/usr/local/cuda-11.0/bin/bin/nvcc': No such file or directory Have you any idea about this issue? |
Before I stuck like you also, make sure you delete everything before you install Cuda, PyTorch, mish-cuda |
The issue was solved. I re-installed the cuda 11.0 and correspoding cudnn. |
Install the program to the running build_ ext, this error occurred:[WinError 2] The system cannot find the specified file. |
Just use yolov7/8 ?
…________________________________
From: hoangduyloc ***@***.***>
Sent: 30 January 2021 3:58 PM
To: WongKinYiu/ScaledYOLOv4 ***@***.***>
Cc: Subscribed ***@***.***>
Subject: Re: [WongKinYiu/ScaledYOLOv4] Can't use mish-cuda in cuda 11.1 (#119)
yes, add environment variable:export TORCH_CUDA_ARCH_LIST="7.5" downgrade version
it works!
Hi there, I stuck install mish-cuda in RTX3090 too
export TORCH_CUDA_ARCH_LIST="7.5" downgrade version I don't really get your mean
Can you please share with me the steps in details, My Cuda 11.0, RTX3090
Thank you~
—
You are receiving this because you are subscribed to this thread.
Reply to this email directly, view it on GitHub<#119 (comment)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/AKPUVMSUZYGEDOH6BFPNOETS4QUCBANCNFSM4VXKHYLA>.
|
Hi, My GPU is RTX3090, it's cuda version is 11.1,so I can't install mish-cuda,is there another way to install mish-cuda?
or use pytorch mish function:
class Mish(nn.Module):
def init(self):
super().init()
print("Mish activation loaded...")
def forward(self,x):
x = x * (torch.tanh(F.softplus(x)))
return x
but I don't know whether speed would be slower than mish-cuda in inference stage?such as FPS
The text was updated successfully, but these errors were encountered: