nvidia-smi和nvcc -V下面显示的cuda版本不一致

系统: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了

你可能感兴趣的:(python,linux,深度学习,pytorch)