1.Anaconda、CUDA及cuDNN
参考上篇博客:Anaconda、TensorFlow安装及环境配置
2.补充:NVIDIA GPU显卡驱动安装
#禁用桌面管理环境lightdm,如果没安装lightdm,Ubuntu默认使用的时gdm3
sudo service lightdm stop
卸载可能存在的旧版本NVIDIA驱动(不执行,重启后会存在问题)
sudo apt-get --purge remove nvidia*
安装驱动需要的依赖
sudo apt-get update
sudo apt-get install dkms build-essential Linux-headers-generic
把nouveau驱动加入黑名单并禁用nouveau内核模块
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在blacklist-nouveau.conf文末添加:
blacklist nouveau
options nouveau modeset=0
保存退出后执行:
sudo update-initramfs -u
重启:
reboot
3.下载并安装PyTorch
在PyTorch官网,找到下载的版本:
conda install pytorch torchvision torchaudio cudatoolkit=10.0
或指定版本然后安装:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.0
4.验证是否安装成功(GPU版本)
import torch
torch.__version__ #输出torch版本
torch.cuda.is_available() #返回True,则表示GPU可用