1. 升级pip
遇到AttributeError: 'NoneType' object has no attribute 'bytes'错误
使用以下代码解决
easy_install -U pip
2. 安装pytorch
查看CUDA版本
进入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)
成功!