Ubuntu中安装GPU版本的pytorch/windows下安装cpu版本的pytorch

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之后就可以用了