快捷導(dǎo)航

Win10+GPU版Pytorch1.1安裝的安裝步驟

更新時(shí)間：2019年09月27日 14:22:30 作者：Taosiyu

這篇文章主要介紹了Win10+GPU版Pytorch1.1安裝的安裝步驟，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

安裝cuda

更新nvidia驅(qū)動(dòng)

打開(kāi)GeForce Game Ready Driver或在GeForce Experience中下載符合自己gpu的程序。

選擇cuda

打開(kāi)nvidia控制面板

點(diǎn)擊幫助、點(diǎn)擊系統(tǒng)信息、在點(diǎn)擊組件在3D設(shè)置中可以看到cuda信息

在我升級(jí)過(guò)nvidia驅(qū)動(dòng)后，cuda的版本更新到了10.1。接下來(lái)下載cuda .

cuda10.1安裝完畢。

安裝cuDNN

在安裝了cuda10.1后選擇對(duì)應(yīng)的cuDNN版本v7.6.1

解壓文件，然后添加bin目錄到環(huán)境變量

安裝cuDNN

在安裝了cuda10.1后選擇對(duì)應(yīng)的cuDNN版本v7.6.1

解壓文件，然后添加bin目錄到環(huán)境變量

使用Anaconda prompt創(chuàng)建環(huán)境

conda create -n pytorch_gpu pip python=3.7

其中pytorch_gpu是環(huán)境的名字，python版本是3.7

conda activate pytorch_gpu

這個(gè)命令可以進(jìn)入創(chuàng)建好的環(huán)境。

安裝Pytorch

建議切換源，我使用的是中國(guó)科學(xué)技術(shù)大學(xué)”（USTC Open Source Software Mirror）提供anaconda的鏡像源，速度提高不少。

方法一：

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

方法二：

打開(kāi).condarc文件，進(jìn)行編輯

這個(gè)順序存在一個(gè)優(yōu)先級(jí)，會(huì)先在中國(guó)科學(xué)技術(shù)大學(xué)的鏡像源尋找，如果大家感覺(jué)速度慢，也可以換別的鏡像源。

打開(kāi)官網(wǎng)選擇后復(fù)制命令

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

整個(gè)過(guò)程只需要按一下'y'即可。

驗(yàn)證

為了確保PyTorch是否安裝成功，我們需要運(yùn)行簡(jiǎn)單的樣例代碼測(cè)試，例如打印出隨機(jī)生成的張量矩陣，以及gpu是否可以使用。

import torch
x = torch.rand(5,5)
print(x)

輸出類似下面

tensor([[0.7078, 0.1424, 0.3411, 0.3987, 0.3476],
    [0.7534, 0.7137, 0.3489, 0.4226, 0.3640],
    [0.4104, 0.8411, 0.5112, 0.0629, 0.0664],
    [0.7568, 0.9495, 0.3300, 0.2392, 0.6441],
    [0.7615, 0.1883, 0.6001, 0.9663, 0.3313]])

如果ok，pytorch安裝成功。