- 首先確認顯卡情況
找到顯卡驅動,在系統信息中找到對應的CUDA版本
- 去官網下載安裝CUDA和cudnn(參考博文)
CUDA:https://developer.nvidia.com/cuda-downloads
cudnn:https://developer.nvidia.com/rdp/cudnn-download
-
安裝Python
需要看清楚python是32位還是64位,目前大部分新版本的Pytorch都只有64位的,所以python要安裝64位的。這裏選的是py3.7 -
安裝pytorch
許多文章介紹的都是用pip和conda,然而我試過了都不行(沒有1.4.0版本的),最後去官網目錄挨個找對應版本的下載,目前沒有cuda10.2的,最多10.1,就下101的,python版本和位數要對應(如果不對應,會無法安裝成功,即使改了名字安裝成功了,也會報import torch._C的錯誤):
然後直接 pip install torch……………… 即可安裝成功(需要先安裝numpy)。