conda list
查找安装的pytorch发现是cpu版本,gpu版本会有类似+cu113
的后缀,表示cuda11.3的gpu版本pytorch- 在运行mmrotate代码python setup.py develop时有时遇到找不到nvcc的错误,使用
nvcc -V
出现错误,使用torch.cuda.is_available()
出现false
,最终发现错误路径是:usr/local/cuda
,且usr/local
确实仅有cuda-11.3
,但其实usr/local/cuda
是一个软连接,自动导向usr/local/cuda-11.3
,最终发现错误是所给错误路径为:usr/local/cuda
,多了一个冒号,但由于linux系统需要带有冒号的路径,所以可以采用一下代码暂时改变
export CUDA_HOME=usr/local/cuda
- 运行别人的mmrotate代码(不直接来自于官方)中新加入模型时出现报错
KeyError: “ is not in the models registry
解决:python setup.py develop