ubuntu18.04下安装pytorch-gpu版
前面
之前介绍过TensorFlow-gpu版本的安装过程。现在再说说pytorch的安装过程。
1.安装Ubuntu后，先安装nvida显卡驱动。
2.安装anaconda。
3.安装cuda。
4.安装cudnn
5.安装pytorch-gpu
前面5个步骤 参考 TensorFlow的安装过程中。
现在主要介绍下pytorch-gpu的安装。
5、安装pytorch-gpu
进入官网
选择对应的版本。
默认用的是conda下载。这里比较坑，直接下载安装几乎下不动，我用了国内镜像服务器，居然发现没有对应的版本。
纠结再三，选择了pip下载安装。