win10安裝pytorch1.4.0過程記錄

  1. 首先確認顯卡情況
    找到顯卡驅動,在系統信息中找到對應的CUDA版本

在這裏插入圖片描述

  1. 去官網下載安裝CUDA和cudnn(參考博文

CUDA:https://developer.nvidia.com/cuda-downloads
cudnn:https://developer.nvidia.com/rdp/cudnn-download

  1. 安裝Python
    需要看清楚python是32位還是64位,目前大部分新版本的Pytorch都只有64位的,所以python要安裝64位的。這裏選的是py3.7

  2. 安裝pytorch
    許多文章介紹的都是用pip和conda,然而我試過了都不行(沒有1.4.0版本的),最後去官網目錄挨個找對應版本的下載,目前沒有cuda10.2的,最多10.1,就下101的,python版本和位數要對應(如果不對應,會無法安裝成功,即使改了名字安裝成功了,也會報import torch._C的錯誤):
    在這裏插入圖片描述
    然後直接 pip install torch……………… 即可安裝成功(需要先安裝numpy)。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章