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

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

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

檢查自己的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那一行。如下圖所示:


在這里插入圖片描述

點擊進入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)上下載以前版本的文件。


在這里插入圖片描述

點擊 Previous PyTorch Versions


在這里插入圖片描述

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

驗證你的pytorch安裝好沒有

在這里插入圖片描述

這樣就安裝好了。

python
import torch
exit()

恭喜你安裝成功。

總結(jié)

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

相關(guān)文章

最新評論

东山县| 永福县| 安阳市| 景德镇市| 楚雄市| 鹤峰县| 合阳县| 当涂县| 长丰县| 琼海市| 乐东| 德江县| 阜平县| 麻城市| 宁明县| 庆城县| 同德县| 镇康县| 全南县| 峨边| 商河县| 台南县| 南城县| 牡丹江市| 西宁市| 汶川县| 天全县| 涞水县| 洛宁县| 东丰县| 阿瓦提县| 杂多县| 繁峙县| 堆龙德庆县| 通化县| 汤阴县| 雷州市| 铁岭市| 武山县| 平度市| 台山市|