Ubuntu 16.04 pip 安装 Pytorch 0.4.1

蔚落 2022-03-22 15:49 227阅读 0赞

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

参考：[https://blog.csdn.net/gdymind/article/details/82153643][https_blog.csdn.net_gdymind_article_details_82153643]

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

于是我在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

说明安装成功。

[https_blog.csdn.net_gdymind_article_details_82153643]: https://blog.csdn.net/gdymind/article/details/82153643
[Link 1]: https://blog.csdn.net/zywvvd/article/details/86355060