安装Pytorch2.0.1+cuda11.7踩雷历程

换源问题

打算跑一下yolov10,我看的教程在安装pytorch的时候强调了不要换源直接下(可能导致调用不了GPU)。但是我直接pip默认用了清华源,所以恢复了一下默认源。

版本问题

我自己的电脑是cuda11.7,去pytorch官网找对应的下载代码是这样

但是下载成功之后执行torch.__version__显示是cpu,也就是根本调用不了gpu。

PS:用conda下载的时候Solving Environment一直转圈

解决方法

pip3 install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2+cu117 -f https://download.pytorch.org/whl/cu117/torch_stable.html

安装成功!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值