在Linux服务器上安装Pytorch1.5.1版本

Zpadger

于 2020-12-08 16:06:18 发布

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏：踩坑记录文章标签： linux pytorch pip

本文链接：https://blog.csdn.net/SpadgerZ/article/details/110875370

本文介绍了如何在Linux服务器上安装Pytorch1.5.1版本，针对GitHub项目环境要求，避免默认安装的1.7.0版本。通过使用豆瓣源，安装过程快速高效，并验证了CUDA支持及Pytorch版本。该步骤对于后续学习yolov5的环境配置至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在跑Github某个项目的时候，发现环境配置要求

Python >= 3.7
Pytorch == 1.5.x

目前官网上安装，默认是1.7.0版本的

安装Pytorch1.5.1版本指令

pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

采用豆瓣源安装

速度还是挺快的

安装完成！

查看一下是否支持CUDA和Pytorch版本号

接下来学习yolov5可能会用到

史上最详细yolov5环境配置搭建+配置所需文件

最低0.47元/天解锁文章

200万优质内容无限畅学