Ubuntu 16.04 LTS 安装PyTorch

Ubuntu 16.04 LTS 安装PyTorch

• 1.安装
• 2.测试
• 3.其它问题
• 4.参考

1.安装

1.打开终端
2.运行命令(由于我的CUDA版本为9.0，所以安装1.1.0版本的PyTorch, 0.3.0版本的torchvision)

pip install torch==1.1.0 torchvision==0.3.0 --user

安装完成后分别显示：

Successfully installed torch-1.1.0 torchvision-0.3.0

这里，需要提醒大家：
1.需要先安装CUDA，且CUDA和PyTorch以及PyTorch和torchvision之间的版本对应关系必须正确
2.官网给出的部分安装命令有问题，比如下图，安装时并不是叫pytorch而是torch。

2.测试

1.打开终端
2.运行命令

python

输入：

import torch
import torchvision

此时不报错，即为安装成功。
还可以进一步查看PyTorch的版本，如下：

fire@wargod:~$ python 
Python 2.7.12 (default, Oct  8 2019, 14:14:10) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch 
>>> import torchvision
>>> print(torch.__version__)
1.1.0

3.其它问题

1.问题

ImportError: No module named builtins

解决方法：

pip install future --user

4.参考

• PyTorch官网

来源：CSDN

作者：人生之疆

链接：https://blog.csdn.net/qq_33547243/article/details/104360058