Skip to content

Latest commit

 

History

History
35 lines (20 loc) · 1.36 KB

README.md

File metadata and controls

35 lines (20 loc) · 1.36 KB

pytorch安裝GPU

一.先查看cuda版本

1.桌面>右鍵>NVIDIA控制面板

Untitled

2.點擊"系統資訊”

Untitled (1)

3.點擊"元素”,並查看第三行:NVIDIA CUDA 11.6.127 driver

image

或者在 prompt 中輸入 nvidia-smi

image

二、 1.到這裡去下載對應的.whl(有兩個要下載!),python的版本若為3.7.多少,那下面的whl檔,就選cp37

1 Ctrl+f尋找(打你cuda的版本,我是11 7,就打cu11)

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