win10安裝pytorch1.4.0過程記錄

首先確認顯卡情況 找到顯卡驅動,在系統信息中找到對應的CUDA版本 去官網下載安裝CUDA和cudnn(參考博文) CUDA:https://developer.nvidia.com/cuda-downloads cudnn:https://developer.nvidia.com/rdp/cudnn-download 安裝Python 需要看清楚python是32位還是64位,目前大部分新版本的
相關文章
相關標籤/搜索