PyTorch----安装

Posted 2020-10-23 smile-sunshine

 由于PyTorch官网 http://pytorch.org/ 推荐使用Anaconda作为软件包管理器（package manager），所以我们需要首先准备好Anaconda环境。

1.准备好Anaconda环境

具体参见：http://blog.csdn.net/zhdgk19871218/article/details/46502637

2.进入PyTorch官网，选择自己相应的版本

 查看python版本

1 fenglei@gpu01:~$ python -V
2 Python 3.5.2 :: Anaconda 4.2.0 (64-bit)

查看cuda版本

1 fenglei@gpu01:~$ cat /usr/local/cuda/version.txt
2 CUDA Version 8.0.61

注意：确保拥有最新的pip 和 numpy packages.

1 fenglei@gpu01:~$ pip install --upgrade pip
2 Collecting pip
3   Downloading pip-9.0.1-py2.py3-none-any.whl (1.3MB)
4     100% |████████████████████████████████| 1.3MB 13kB/s
5 Installing collected packages: pip
6   Found existing installation: pip 8.1.2
7     Uninstalling pip-8.1.2:
8       Successfully uninstalled pip-8.1.2

 

1 fenglei@gpu01:~$ pip install --upgrade numpy
2 Requirement already up-to-date: numpy in ./anaconda3/lib/python3.5/site-packages

3.根据Run this command的提示，安装PyTorch

如果下载速度太慢的话，在自己的Linux系统上可以在浏览器里新建任务，下载链接就是安装命令中第一行的网址，把这个whl文件下载下来；在Windows上可以用迅雷下载，新建任务进行下载。

下载完成之后拷贝到自己的Linux系统上去或者服务器上，然后找到文件位置，直接pip install 文件名。

安装torch：

安装成功：

安装torchvision：

安装成功：

4.验证PyTorch是否安装成功

1 fenglei@gpu01:~$ python
2 Python 3.5.2 |Anaconda 4.2.0 (64-bit)| (default, Jul  2 2016, 17:53:06)
3 [GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux
4 Type "help", "copyright", "credits" or "license" for more information.
5 >>> import torch
6 >>> import torchvision
7 >>>

不报错的话就说明安装成功。