一、我的環(huán)境
- 我的顯卡是RTX4050鸽疾，首先去官網(wǎng)下載CUDA Toolkit
https://developer.nvidia.com/cuda-downloads
我是12.3version弛姜，下載并安裝
二乙各、下載pytorch
-
大部分的參考博主都推薦的是官網(wǎng)下載
但是我從官網(wǎng)下載的，不知道為什么總是下的cpu版本
image.png
雖然是按照教程來的炼蛤，選的是gpu12.1猿挚，但是結(jié)果還是False
- 我的另一個方法（參考某個博主https://cloud.tencent.com/developer/article/2022050）：
- 從這里下載：https://download.pytorch.org/whl/torch_stable.html
- 選取和你匹配的version
- 我的python：3.11咐旧，CUDA Version: 12.3
- 需要下載：cu121/torch-2.1.1%2Bcu121-cp311-cp311-win_amd64.whl和cu121/torchvision-0.16.1%2Bcu121-cp311-cp311-win_amd64.whl
- 然后就通過conda prompt進入文件放置的位置
- pip install xxx.whl
- 記住先安裝pytorch再安裝torchversion
三、查看torch是否安裝成功gpu版本
import torch
print(torch.cuda.is_available())
如果是True就是成功了
四绩蜻、查看自己的nVidia狀態(tài)
- cmd 輸入：nvidia-smi
