anaconda中使用pip安装pytorch(GPU)

这篇博客介绍了如何在已经安装了CUDA和cuDNN的环境下,通过pip在Anaconda中安装PyTorch的GPU版本。首先确认CUDA版本为11.3,然后从PyTorch官网找到对应配置并复制安装命令,激活目标环境后执行安装。最后通过Python代码验证安装成功,`torch.cuda.is_available()`返回True。

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

anaconda中使用pip安装pytorch(GPU)

由于conda直接安装torch遇到了问题,在此记录一下另一种装法。

一、确认cuda与cudnn安装完成

我在以前已经安装完成了 在这仅确认一下cuda版本即可,cmd中打入:

nvcc --version

image-20220806175350107

可见是11.3版本

二、下载pytorch

pytorch官网

image-20220806175521302

找到对应的安装配置,复制链接,到conda中激活要安装的环境,直接粘贴后回车。

image-20220806180327136

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-nETJSa5c-1659781401625)(C:\Users\13587\AppData\Roaming\Typora\typora-user-images\image-20220806180354424.png)]

检查一下

import torch
torch.cuda.is_available()

返回True,安装成功!

相关conda命令:

conda create --name envname python=3.8 -y
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值