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
文章出处登录后可见！
已经登录？立即刷新
