由于 PyTorch 官网 http://pytorch.org/ 推荐使用 Anaconda 作为软件包管理器 (package manager), 所以我们需要首先准备好 Anaconda 环境
1. 准备好 Anaconda 环境
具体参见: http://blog.csdn.net/zhdgk19871218/article/details/46502637
2. 进入 PyTorch 官网, 选择自己相应的版本
查看 python 版本
- fenglei@gpu01:~$ python -V
- Python 3.5.2 :: Anaconda 4.2.0 (64-bit)
查看 cuda 版本
- fenglei@gpu01:~$ cat /usr/local/cuda/version.txt
- CUDA Version 8.0.61
注意: 确保拥有最新的 pip 和 numpy packages.
- fenglei@gpu01:~$ pip install --upgrade pip
- Collecting pip
- Downloading pip-9.0.1-py2.py3-none-any.whl (1.3MB)
- 100% || 1.3MB 13kB/s
- Installing collected packages: pip
- Found existing installation: pip 8.1.2
- Uninstalling pip-8.1.2:
- Successfully uninstalled pip-8.1.2
- fenglei@gpu01:~$ pip install --upgrade numpy
- Requirement already up-to-date: numpy in ./anaconda3/lib/python3.5/site-packages
3. 根据 Run this command 的提示, 安装 PyTorch
如果下载速度太慢的话, 在自己的 Linux 系统上可以在浏览器里新建任务, 下载链接就是安装命令中第一行的网址, 把这个 whl 文件下载下来; 在 Windows 上可以用迅雷下载, 新建任务进行下载
下载完成之后拷贝到自己的 Linux 系统上去或者服务器上, 然后找到文件位置, 直接 pip install 文件名
4. 验证 PyTorch 是否安装成功
- fenglei@gpu01:~$ python
- Python 3.5.2 |Anaconda 4.2.0 (64-bit)| (default, Jul 2 2016, 17:53:06)
- [GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux
- Type "help", "copyright", "credits" or "license" for more information.
- >>> import torch
- >>> import torchvision
- >>>
不报错的话就说明安装成功
来源: http://www.bubuko.com/infodetail-2490697.html