1.安装显卡驱动
首先查看自己适合安装的驱动
$:sudo add-apt-repository ppa:graphics-drivers/ppa
$:sudo apt-get update
$:sudo apt-get install nvidia-384
$:sudo reboot
$:sudo nvidia-smi
如果显示GPU列表,则证明驱动安装成功了
2.安装cuda
在官网中找到下载文件:选择对应版本(见下图cuda10)并点击Download按钮:
下载完成后根据上述命令进行安装
3.pytorch-gpu安装
打开anaconda-navigator,搜索package,完成后点击apply(这里要注意一下,python3.5版本以上的pytorch-gpu没有,需要在anaconda中创建一个python3.5的环境,之后验证gpu可用,torch.cuda.is_available()为true)。
windows下安装pytorch
打开anaconda navigator,搜索torch,进行下载安装,这里推荐使用清华镜像网址(在channel中添加:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/),
安装完成后可以再用同样的方法安装torchvision之后就可以用了