pytorch安装及其版本切换

本文详细介绍了如何安装PyTorch1.4,包括检查CUDA版本,下载并安装CUDA10.1,配置环境变量,创建conda虚拟环境安装PyTorch,以及处理安装问题和验证安装成功。同时提到了版本切换的方法,强调了PyTorch与CUDA之间的版本对应关系。

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

一、安装说明

pytorch是动态库,依赖的库各个版本的pytorch可能是不同的,所以pytorch和依赖库有版本对应关系。如下:
https://cloud.tencent.com/developer/article/2038557
安装pytorch需要提前安装对应的cuda本版,具体版本查询上面的链接。(安装cuda时要注意driver version的要求,不然不能运行。查看driver version可以在NVIDIA的控制面板查看)。

二、安装步骤

这里我需要下载安装pytorch 1.4

1. 下载安装cuda

(1)查看版本

在这里插入图片描述
1.4版本的pytorch需要10.1的cuda驱动。

(2)下载

cuda历代版本下载

选择对应版本进入下载页面(推荐local下载,如果安装失败不需要重新下载,节省流量和时间)在这里插入图片描述

(3)安装

在这里插入图片描述
在这里插入图片描述在这里插入图片描述
安装完成(这里我以前安装了10.0的版本,所以有个v10.0)
在这里插入图片描述

(4)环境变量配置

安装完默认添加CUDA_PATH和CUDA PATH_V10_1的环境变量
在这里插入图片描述
在path环境变量中也有添加在这里插入图片描述
到此cuda安装完成

2. pytorch的安装

pytorch的安装较为简单,可以直接用命令行安装
用conda新建一个虚拟环境便于pytorch版本的管理

conda create -n pytorch14 python=3.7

激活环境

activate pytorch14

安装pytorch
命令行在官网进行查询
pytorch安装命令行
在这里插入图片描述
输入命令行即可安装

报错处理

在这里插入图片描述
安装出现错误,可以自行下载whl文件,通过pip安装。
参考https://blog.csdn.net/candice5566/article/details/104929677
https://blog.csdn.net/weixin_43332715/article/details/123518061
下载地址

验证

在这里插入图片描述
在这里插入图片描述
输出True说明安装成功

三、本版切换

因为pytorch和cuda是版本对应的,所以切换pytorch版本需要切换对应的cuda版本。

1. 切换cuda环境变量
2. 切换pytorch对应的虚拟环境
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

wuming先生

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值