【python环境之torch】为现有环境安装pytorch、torchvision和torchaudio

本文详细介绍了在已安装CUDA和显卡驱动的现有环境中,针对cu117和python3.9版本,通过pip、离线安装和不推荐的conda方法安装torch、torchvision和torchaudio的步骤。重点强调了在选择安装包时的注意事项和验证安装成功的步骤。

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

【深度学习环境之torch】为现有环境安装torch、torchvision、torchaudio,以cu117+python3.9为例

由于大多数帖子在编写安装教程时都是从安装显卡驱动和cuda开始。但是在现实应用中,大多数情况下电脑都已经安装完cuda和显卡驱动且代码需要在固定python版本下运行。
针对以上问题,本篇以cu117+python3.9为例介绍如何为现有环境安装pytorch等python库。本篇共介绍三种安装方式,分别是,pip安装、离线安装和conda安装

总述(重要!!重要!!重要!!!)

由于torch与操作系统、python版本和cuda版本有关 ,所以无论是pip安装、离线安装还是conda安装,安装的流程均为

查看cuda版本->根据对应版本查找pytorch对应文件->下载安装包并安装

1、pip安装

作为常用的安装pytorch 的方法,使用pip安装主要包含以下几步:

1、查看电脑上的cuda版本(以linux系统为例)

使用如下指令查看自己电脑上的cuda版本:

nvcc-V

输入后出现以下界面,图中红框中的内容为电脑的cuda版本。
在这里插入图片描述

注意!!!!!!!在实际应用中,常使用如下指令查看电脑显卡的运行情况

nvidia-smi

在这里插入图片描述

图中标红处也显示cuda版本。但是,此cuda版本并不是系统安装的cuda版本,

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值