系统:Linux
显卡:3090
nvidia-smi和nvcc -V下面显示的cuda版本不一致,(nvcc -V:10.1,nvidia -smi:11.4)导致安装的pytorch是cuda10.1的,导致3系列显卡不能调用。
解决方法:
vim/gedit ~/.bashrc
利用编辑器修改bashrc,在最后添加CUDA的路径
export LD_LABRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64
export PATH=$PATH:/usr/local/cuda-11.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.2
export PATH=/usr/local/cuda/bin:$PATH
激活bashrc
source ~/.bashrc
然后使用nvcc -V就可以看到此时的CUDA已经变成了11.X了
版权声明:本文为qq_42964728原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。