Ubuntu安装torch和torchvision

最新推荐文章于 2025-05-25 16:47:44 发布

WaitFoF

最新推荐文章于 2025-05-25 16:47:44 发布

阅读量1.5w

点赞数 8

CC 4.0 BY-SA版权

分类专栏：软件安装/环境配置文章标签： torchvision ubuntu pytorch

没有太晚的开始，不如就从今天行动。总有一天，那个一点一点可见的未来，会在你心里，也在你的脚下慢慢清透。生活，从不亏待每一个努力向上的人。

本文链接：https://blog.csdn.net/qq_41498261/article/details/109752013

一、 cpu版本

1. 安装torch

经过安装的经验，发现应该先安装torchvision。

1、下载
下载whl类型的文件：torch

cu表示的是CUDA的版本，cp表示的是python的版本。

2、安装pytorch

将下载的文件放入home下，打开终端进行安装：

pip3 install torch-0.4.0-cp35-cp35m-linux_x86_64.whl

wf@ubuntu:~/Downloads$ pip install  torch-0.4.0-cp36-cp36m-linux_x86_64.whl 
Processing ./torch-0.4.0-cp36-cp36m-linux_x86_64.whl
distributed 1.21.8 requires msgpack, which is not installed.
Installing collected packages: torch
Successfully installed torch-0.4.0

2. 安装torchvision

1、使用pip命令安装

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple   torchvision

wf@ubuntu:~/Downloads$ pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple   torchvision
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting torchvision
  Downloading https://pypi.tuna.tsinghua.edu.cn/packages/e7/43/aaa740c406b1832adc6ff9d5e71c23fd2af2ebd436c42d76d85809ec8be9/torchvision-0.8.1-cp36-cp36m-manylinux1_x86_64.whl (12.8 MB)
     |████████████████████████████████| 12.8 MB 609 kB/s 
Requirement already satisfied: numpy in /home/wf/anaconda3/lib/python3.6/site-packages (from torchvision) (1.14.3)
Collecting torch==1.7.0
  Downloading https://pypi.tuna.tsinghua.edu.cn/packages/80/2a/58f8078744e0408619c63148f7a2e8e48cf007e4146b74d4bb67c56d161b/torch-1.7.0-cp36-cp36m-manylinux1_x86_64.whl (776.7 MB)
     |████████████████████████████████| 776.7 MB 180 bytes/s 
Requirement already satisfied: pillow>=4.1.1 in

最低0.47元/天解锁文章