安装pytorch纯小白教程||重点是要注意版本!

打开pytorch安装网址,根据系统配置选择特定版本

在Terminal输入命令,查看CUDA Version

nvidia-smi

 e.g.我的CUDA版本是11.2,下载CUDA11.8可向下兼容,因此对应我的操作系统这样选择

复制下面 Command 到 Terminal 中运行

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 完成后运行以下代码进行测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())

成功后最后一个会显示true

注意!

一开始其实是false,于是conda list了一下发现,pytorch-mutex是cpu版本,但是不管如何uninstall还是按照其他方法重下,重安再卸cpuonly,还是不行。索性把环境删了冲装,这次我装的python版本是3.9(上一次是3.7),于是奇迹般地True了!

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

一天只能上一天班

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

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

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

打赏作者

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

抵扣说明:

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

余额充值