以下步骤参考:
感谢各位优秀博主的资源提供,此处这是字节配置环境的一些记录!!!
1.显卡驱动
直接通过ubuntu图形界面Software&updates - additionnal drivers 选择之后apply即可;之后重启电脑通过nvidia-smi 查看是否安装成功即可;
2.cuda安装
此处根据自己的电脑显卡驱动版本对应即可,可以实现向下兼容
注意:1.解压安装包命令,注意对应好自己版本tar -xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
2.安装之后,在更新环境变量的过车中 如果出现nvcc -V 命令无法找到 ----则需更新地址变 量
3.除了在/.bashrc中配置LD_LIBRARY_PATH路径以外,还需要配置PATH路径,完整配置如下:
export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$PATH:/usr/local/cuda/bin
3.cudnn安装
按照对应版本关系选择cudnn版本进入官网下载,https://developer.nvidia.com/rdp/cudnn-downloadhttps://developer.nvidia.com/rdp/cudnn-download注意:若出现无法获取’cuda/include/cudnn.h’
解决方案:在解压文件夹目录打开终端依次输入
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include #这里*很关键
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
检测安装成功:cat cuda/include/cudnn_version.h |grep ^#
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2均可
3.Anaconda安装
具体安装版本及对应链接
Old package lists — Anaconda documentationhttps://docs.anaconda.com/anaconda/packages/oldpkglists/
提供国内源的anaconda下载,此处提供的是 Anaconda3-2021.11-Linux-x86_64.shhttps://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2021.11-Linux-x86_64.shhttps://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2021.11-Linux-x86_64.sh
安装命令更改:chmod +x Anaconda3-2021.11-Linux-x86_64.sh
bash Anaconda3-2021.11-Linux-x86_64.sh
注意:在安装包现在的文件夹里面打开终端就行,之后后续步骤按照开头多提供的链接一次进行就可;并且输入python检测是否安装成功之后,通过ctrl+d 退出编译环境(高手忽略即可);
4.Pytorch安装
选择对应版本,安装最好直接从官网安装,如果由于网速原因导致安装某个包失败的话就重复conda命令多安装几次,他会自动判断附件包是否已经安装过了。
检测是否安装完成:依次输入
python
import torch
torch.__version__
torch.cuda.is_available()
如果安装完成输出如下
ok!!!
5. 除了上述的验证方式,有的用户也可以通过命令行进入python环境依次输入
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
显示结果
如果依次显示证明安装成功!!!