最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

win10使用清華源快速安裝pytorch-GPU版(推薦)

 更新時間:2019年10月24日 11:20:52   作者:來自星星的菜鳥  
這篇文章主要介紹了win10使用清華源快速安裝pytorch-GPU版,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價值,需要的朋友可以參考下

檢查自己的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 文件,打開文件刪除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時會把torchvision也給安裝上,就不用后面再安裝torchvision了。cudatoolkit=10.1表示cuda版本10.1。-c表示使用官網(wǎng)下載。不含-cpu表示gpu版。

配置自己cuda版本的pytorch,cuda版本過低怎么辦

如上所說,直接將cudatoolkit=10.1改為你自己的cuda版本,如我的是cuda 9.0:


在這里插入圖片描述

conda install pytorch torchvision cudatoolkit=9.0 pytorch

安裝就行了。

安裝指定版本的torchvision

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


在這里插入圖片描述

點(diǎn)擊 Previous PyTorch Versions


在這里插入圖片描述

上面就有你想要找到的版本對應(yīng)文件,下載下來 pip install就好了。需要注意很多版本問題,如:cuda9.0的torchvison只支持到0.3.0版本等等。

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

在這里插入圖片描述

這樣就安裝好了。

python
import torch
exit()

恭喜你安裝成功。

總結(jié)

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

相關(guān)文章

最新評論

湘阴县| 勃利县| 英山县| 监利县| 凤城市| 闽清县| 安康市| 西乡县| 榕江县| 陇南市| 游戏| 泗洪县| 汪清县| 米易县| 罗定市| 大邑县| 来安县| 贵定县| 田东县| 嘉峪关市| 桐庐县| 土默特左旗| 安宁市| 肇庆市| 丹江口市| 陇川县| 崇义县| 沾益县| 珲春市| 秀山| 肇庆市| 乌兰浩特市| 阿勒泰市| 长治县| 武穴市| 芦山县| 呼伦贝尔市| 白玉县| 上杭县| 柯坪县| 尉氏县|