如果有帮助,记得回来点个赞
目录
- 1.安装指定GPU驱动
-
- [如果安装的GPU CUDA Version和CUDA Toolkit版本已经冲突怎么办?](#如果安装的GPU CUDA Version和CUDA Toolkit版本已经冲突怎么办?)
- [2.安装指定版本的CUDA Toolkit](#2.安装指定版本的CUDA Toolkit)
-
- [如果我安装了CUDA Toolkit之后nvcc -V仍然显示旧的CUDA Toolkit版本怎么办?](#如果我安装了CUDA Toolkit之后nvcc -V仍然显示旧的CUDA Toolkit版本怎么办?)
- 3.安装与CUDA对应的Pytorch
1.安装指定GPU驱动
在安装指定GPU驱动的时候,一定要注意一个点,就是你的CUDA Driver Version一定要和CUDA Toolkit保持一致.也就是说
bash
nvidia-smi
和
bash
nvcc -V
的版本号要保持一致.如下图所示
所以如果你的GPU如果没有驱动也没有CUDA tool kit的话,可以先选择只安装GPU驱动,然后在自己指定的虚拟环境中再安装指定版本的CUDA tool kit,也就是说GPU驱动是千万不能装错的,一旦GPU装好了,一定一定不要乱动它,否则直接炒蛋.能给人整爆炸!!!
其中CUDA tool kit选择X,表示不进行这个安装!!
如果安装的GPU CUDA Version和CUDA Toolkit版本已经冲突怎么办?
有办法解决!
首先GPU CUDA Version和GPU驱动一般都是最合适的,就是下载GPU驱动就有了一个比较合适的CUDA Driver Version了,然后一定要选择英文版本的GPU驱动程序,不要选择中文的!
接着在选择CUDA Toolkit版本的时候,直接参考本博客第二章内容就行👇
2.安装指定版本的CUDA Toolkit
具体安装方法是根据上面这个链接里面的Installation Instruction指令进行安装!
务必记住一点:在Linux环境中,比如是ubuntu
系统,一定要选择runfile
格式的安装包下载,不要选择deb格式的安装!!
直接在远程SSH窗口中wget他的一个下载链接,之后sudo sh CUDA_XXX.run安装即可!!!
如果我安装了CUDA Toolkit之后nvcc -V仍然显示旧的CUDA Toolkit版本怎么办?
有办法解决!
直接使用gedit或者vim编辑这个文件
bash
vim ~/.bashrc
在这个文件最后一行添加如下环境变量.其中CUDA Toolkit一般就是在这个目录里面,其中如果你安装的是11.7的
CUDA Toolkit的话,目录就是这样的,如果你安装的是10.1版本的话,就把我下面这个命令的11.7换成10.1就行,其他不变!
bash
export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
接着运行命令使环境命令生效
bash
source ~/.bashrc