1. 程式人生 > >Win10如何安裝pytorch1.0+CUDA8.0+cudnn

Win10如何安裝pytorch1.0+CUDA8.0+cudnn

Win10如何安裝pytorch1.0+CUDA8.0+cudnn

Pytorch1.0正式版安裝

1、首先登入官網https://pytorch.org/get-started/locally/,選擇你使用的pytorch版本,本人選擇了pytorch1.0+cuda8.0版本。由於使用conda指令安裝pytorch速度十分慢,因此本人建議選擇pip,下方指令會出現類似於“

https://download.pytorch.org/whl/cu80/torch-1.0.0-cp37-cp37m-win_amd64.whl”的下載地址,然後使用迅雷下載即可。要記住你的下載地址,我的是“G:\torch-1.0.0-cp37-cp37m-win_amd64.whl”
下載完成後,進入cmd介面,輸入如下指令即可完成安裝:pip install G:\torch-1.0.0-cp37-cp37m-win_amd64.whl.
2、安裝好後Pytorch後,還有個torchvision安裝,這個主要集成了一些資料集,深度學習模型,一些轉換等,以後需要使用還是很方便的。使用如下指令安裝:pip install torchvision.

測試是否安裝成功

如果引用torch不報錯,即安裝成功。

安裝Visual C++ 2015 64 bit

進入網址“https://www.microsoft.com/zh-cn/download/details.aspx?id=48145 ”下載,記得x64和x86都要下載,然後點選安裝。如果報錯“已有相應產品安裝”,很大可能是安裝了更高版本的Vc++,比如2017。此時只需將其解除安裝,然後再點選安裝即可。

Cuda Toolkit 8.0安裝

首先進入官網“https://developer.nvidia.com/cuda-downloads”。
然後選擇相應版本,我的選擇為
在這裡插入圖片描述

下面兩個安裝包都要下載(推薦使用迅雷)
在這裡插入圖片描述

下載完成後點選安裝即可。(如果無法點選,只需將其後綴名改為.exe即可)
cuda安裝過程中通常會出現如下問題:在這裡插入圖片描述
解決方案參見該部落格,寫的非常詳細:https://blog.csdn.net/zzpong/article/details/80282814

安裝cudnn

下載地址:https://developer.nvidia.com/rdp/cudnn-archive
解壓後,將bin、include、lib\x64下的檔案分別對應拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit資料夾下的bin、include、lib\x64下。