一、ubuntu16.04
通过镜像安装
二、nvidia驱动
1、 命令行输入 ubuntu-drivers devices 会出现可以安装以及推荐的驱动,后面带recommended就是推荐的版本
2、安装 驱动
sudo apt-get install nvidia-465
3、不过我安装的是384的版本,安装好后输入 nvidia-smi,就会显示驱动状态
三、cuda安装,cuda版本要对应pytorch的版本,我安装的是11.1的cuda版本
进入下载连接,点击版本,选择对应的选项 CUDA Toolkit Archive | NVIDIA Developer
我选择的是11.1的版本 CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA Developer
然后最下面是选择安装方式,我现在的是deb的本地安装方式
按照他给出的命令输入就好了,最后他强行要更新我的显卡驱动到465,行吧行吧,更就更了吧,更新了要重启驱动才能正常
四、pytorch安装
打开下载网址 Start Locally | PyTorch 选择自己的配置,我是安装了anaconda,没有安装可以选择pip安装,好像pytorch有要求python3.6及以上,具体自己测试
然后输入下面的命令
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
此时输入y