檢查自己的cuda是否安裝好

在anaconda prompt中輸入

nvcc -V

顯示如上面表示安裝好了。

配置清華園下載環(huán)境

同樣在在anaconda prompt中輸入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

然后刪除：C:\Users\Administrator 目錄下的 .condarc 文件，打開(kāi)文件刪除defaults那一行。如下圖所示：

點(diǎn)擊進(jìn)入pytorch官網(wǎng)

官網(wǎng)鏈接: https://pytorch.org/get-started/locally/.

自己選擇自己的版本，如下：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

這行代碼的含義是：安裝pytorch時(shí)會(huì)把torchvision也給安裝上，就不用后面再安裝torchvision了。cudatoolkit=10.1表示cuda版本10.1。-c表示使用官網(wǎng)下載。不含-cpu表示gpu版。

配置自己cuda版本的pytorch，cuda版本過(guò)低怎么辦

如上所說(shuō)，直接將cudatoolkit=10.1改為你自己的cuda版本，如我的是cuda 9.0：

conda install pytorch torchvision cudatoolkit=9.0 pytorch

安裝就行了。

安裝指定版本的torchvision

如果你在看代碼的時(shí)候，有些時(shí)候作者要求指定版本的torch或者torchvision，這個(gè)只能在官網(wǎng)上下載以前版本的文件。

點(diǎn)擊 Previous PyTorch Versions

上面就有你想要找到的版本對(duì)應(yīng)文件，下載下來(lái) pip install就好了。需要注意很多版本問(wèn)題，如：cuda9.0的torchvison只支持到0.3.0版本等等。

驗(yàn)證你的pytorch安裝好沒(méi)有

這樣就安裝好了。

python
import torch
exit()

恭喜你安裝成功。

總結(jié)

以上所述是小編給大家介紹的win10使用清華源快速安裝pytorch-GPU版,希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)本站網(wǎng)站的支持！
如果你覺(jué)得本文對(duì)你有幫助，歡迎轉(zhuǎn)載，煩請(qǐng)注明出處，謝謝！

版權(quán)聲明：本站文章來(lái)源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請(qǐng)保持原文完整并注明來(lái)源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來(lái)源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來(lái)，僅供學(xué)習(xí)參考，不代表本站立場(chǎng)，如有內(nèi)容涉嫌侵權(quán)，請(qǐng)聯(lián)系alex-e#qq.com處理。