Ubuntu 16.04 pip 安装 Pytorch 0.4.1

之前用Tensorflow比较多,但是也有不得不实现的Pytorch程序,于是动手安装,指定了版本0.4.1,下面记录我的安装过程。

参考:https://blog.csdn.net/gdymind/article/details/82153643

首先需要安装Ananconda3 和 CUDA,我的安装经验在另一篇博客中有所记录,需要的同学可以移步参考。

于是我在Anaconda3和CUDA9.0的环境下开始了Pytorch 的安装。

官网已经推出了Pytorch1.0的最新版本,而且我不知为何无法登录官网的往期版本界面,于是直接用pip命令安装。

网上很多教程用pip3安装,但我在Anaconda3安装完成之后pip命令已经对应python3.6.8的版本,直接使用pip即可。

pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl

显示信息:

Collecting torch==0.4.1 from http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
  Downloading http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl (519.5MB)
    100% |████████████████████████████████| 519.5MB 7.4MB/s 
Installing collected packages: torch
Successfully installed torch-0.4.1

再查看pip list

多了: torch                              0.4.1    

进入python命令行,输入:

import torch
torch.cuda.is_available()

返回True

继续输入

print(torch.__version__)

返回

0.4.1

说明安装成功。

 

你可能感兴趣的:(Python,Linux)