欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-27 21:07:02
在跑Github某个项目的时候,发现环境配置要求Python >= 3.7Pytorch == 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采用豆瓣源安装速度还是...

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

Python >= 3.7
Pytorch == 1.5.x

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

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

安装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

采用豆瓣源安装

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

速度还是挺快的

安装完成!

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

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

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

接下来学习yolov5可能会用到

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

本文地址:https://blog.csdn.net/SpadgerZ/article/details/110875370