使用
import torch
torch.cuda.is_available()
------> False
print(torch.version) --> 查詢 torch 版本
命令行,輸入 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
選擇該版本後
pip install xx.whl
来源地址:https://blog.csdn.net/chencef/article/details/129107702