登录PyTorch官网:http://pytorch.org/ (ps:官网里面有些链接需要翻墙才可以打开,建议使用火狐里面的v*n插件,方便)

GPU版PyTorch安装

利用conda(官网推荐用Anaconda安装)安装Linux版的的PyTorch,选择Python3.7, CUDA 9.0版本的。因为我的服务器上以及把cuda更新为9.0版本。

GPU版PyTorch安装

之后在Linux终端运行这句话,就可以实现安装。

GPU版PyTorch安装

安装完成后,再运行红色矩形框内的内容进行,验证,看看pytorch是否安装好。

最后再推荐PyTorch中文网:https://ptorch.com/

 

相关文章: