1. nvidia 驱动更新:
sudo apt-get update
sudo add-apt-repository ppa:graphics-drivers/ppa
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall / sudo apt install nvidia-×
2. cuda 更新
2. 安装CUDA 10.0:
(1).从 cuda 官网 下载runfile(local)的CUDA,即cuda_*_linux.run;
(2).执行如下命令,安装CUDA *:
sudo sh cuda_10.0.130_410.48_linux.run
3. 安装pytorch
官网查到相关版本
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch