当前位置:   article > 正文

解決 torch 無法使用GPU_torch is not able to use gpu

torch is not able to use gpu

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

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号