1.使用
import torch
torch.cuda.is_available()
------> False
print(torch.version) --> 查詢 torch 版本

2.命令行,輸入 nvidia-smi
查詢
3.直接去網站找出相對應版本 https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述
cuda : 11.7 -> cu117
python3.9 -> cp39
torch 1.13.0 -> torch-1.13.0
win 64 -> win_amd64

4.選擇該版本後
pip install xx.whl

Logo

加入「COC·上海城市开发者社区」,成就更好的自己!

更多推荐