该文介绍了如何检查CUDA是否可用以及如何找到并安装特定版本的PyTorch。首先，通过`torch.cuda.is_available()`判断GPU是否支持CUDA。接着，运行`nvidia-smi`命令查看GPU信息。然后，根据CUDA版本和Python版本，在PyTorch官网找到对应whl文件。最后，使用`pipinstall`命令安装指定的PyTorch版本。
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
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
1867
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
