当前位置:   article > 正文

Ubuntu16.04离线安装Pytorch_ubutnu pytorch 离线安装

ubutnu pytorch 离线安装

[版本说明]

  • Ubuntu16.04
  • cuda9.0
  • python3.7
  • pytorch-1.1.0
  • torchvision-0.3.0

1. 下载离线包

进入PyTorch官网
在这里插入图片描述
找到自己需要的版本,我需要下载的pytorch版本是1.1.0
在这里插入图片描述
打开这个网址后,根据自己的版本需要,点击下载。
网址上的cu表示该版本支持GPU,1.1.0表示pytorch版本,cp37代表python3.7
在这里插入图片描述

2. 安装torch

(1)进入pytorch安装包的下载目录,打开终端。
(2)创建并激活自己的虚拟环境。

conda create --name wdn python=3.7 # wdn是我的虚拟环境名
source activate wdn
  • 1
  • 2

(3)安装pytorch

pip install torch-1.1.0-cp37-cp37m-linux_x86_64.whl
# torch-1.1.0-cp37-cp37m-linux_x86_64.whl是我下载的whl文件名
  • 1
  • 2

在这里插入图片描述

3. 安装torchvision

torchvision包是服务于pytorch深度学习框架的。安装pytorch时,torchvision独立于torch。必须同时安装了torch和torchvision,才能顺利使用pytorch。
直接再终端输入:pip install torchvision==0.3.0
在这里插入图片描述
注意到图片底端的pillow-7.0.0

4. 验证是否安装成功

打开终端,激活安装了PyTorch和torchvision的虚拟环境,输入python进入python环境
在这里插入图片描述
然后输入

import torch
import torchvision
  • 1
  • 2

结果报错了
在这里插入图片描述
原因就在于刚才第3步安装torchvision中提到的:pillow-7.0.0,安装torchvision时,默认安装的pillow版本为7.0.0,此版本的pillow已经移除了’PILLOW_VERSION’,而pillow6.1还保留着,所以只需要conda install pillow=6.1即可。
在这里插入图片描述
再次验证
在这里插入图片描述
没有报任何错误,说明安装成功。

官网
torchvision:https://pypi.org/project/torchvision/0.3.0/
pytorch:https://pytorch.org/

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/413335
推荐阅读
相关标签
  

闽ICP备14008679号