安装pytorch

1. 升级pip

        遇到AttributeError: 'NoneType' object has no attribute 'bytes'错误

        使用以下代码解决

easy_install -U pip

 2. 安装pytorch

        查看CUDA版本

        

        

进入pytorch下载界面

Start Locally | PyTorch

点击 CUDA12.1进行下载(应该是向下兼容的)

然后复制pip命令在cmd运行

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

不是外网,不用镜像资源,要想快点就开启5G热点下载。

3.测试

import torch
x = torch.rand(5, 3)
print(x)

成功!

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值