参考 最全:在Ubuntu 18安装Nvidia Driver, Cuda 10.2, Pytorch及一些疑问的解答和讨论 - 知乎 和 Ubuntu搭建Pytorch,就这一篇就够了_Chengyunlai的博客-CSDN博客_ubuntu安装pytorch
CUDA 11.4
为了后面顺利进行,确保系统不存在Nvidia Driver,Cuda相关文件。
sudo apt-get purge nvidia*sudo apt --purge remove "cublas*" "cuda*"sudo apt --purge remove libcudnn7 libcudnn7-dev libnccl2sudo apt-get autoremove
安装 Nvidia Driver,通过apt在standard Ubuntu Repository进行driver的安装:
- 参考博文2中,使用图形化安装,我本地出现了无法应用修改,故采用命令行安装。
ubuntu-drivers devicessudo apt install nvidia-driver-470 #此处为的recommnened为470
安装成功后,重启电脑
reboot
查看
nvidia-smi
根据推荐安装11.4 去CUDA Toolkit Archive | NVIDIA Developer
Cuda Toolkit可以通过两种方式安装,一种是distribution-specific packages (即类似与Debian packages), 另一种是distribution-independent package (即runfile packages)。
本文采用了Debian的方式(使用apt)进行了Nvidia driver的安装,因此CUDA也继续采用Debian方式进行安装
安装完成后,输入nvcc --version不显示所安装CUDA的版本号,需要添加以下路径到~/.bashrc(root隐藏文件 ls-a) :
export LD_LIBRARY_PATH=/usr/local/cuda-xx.xx/lib64:$LD_LIBRARY_PATHexport PATH=/usr/local/cuda-xx.xx/bin:$PATHnvcc --version
Pytorch11.4
pytorch11.4 也可安装 11.3,此处pip3安装11.3,参考博文2
注意pip3必须20.0++以上,若不是,先升级pip3
pip3 install --upgrade pip