一.先查看cuda版本
1.桌面>右鍵>NVIDIA控制面板
2.點擊"系統資訊”
3.點擊"元素”,並查看第三行:NVIDIA CUDA 11.6.127 driver
或者在 prompt 中輸入 nvidia-smi
二、 1.到這裡去下載對應的.whl(有兩個要下載!),python的版本若為3.7.多少,那下面的whl檔,就選cp37
2.在到cmd執行一下(要記得切到activate pytorch)
cd C:\Users\17394\Downloads #下載的位置
pip install torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
pip install torchvision-0.8.2+cu110-cp37-cp37m-win_amd64.whl
參考文獻:https://blog.csdn.net/m0_45696735/article/details/118253942