Anaconda安裝Pytorch鏡像詳細(xì)教程

在win10基礎(chǔ)上已經(jīng)安裝好了Anaconda,繼續(xù)安裝Pytorch。因?yàn)椴捎迷镜南螺d方法實(shí)在太慢,還容易出現(xiàn)問題,現(xiàn)采取Pytorch清華鏡像下載方法,親測有效快速?。?!

1.已經(jīng)安裝好Anaconda

2.搜索cmd>>>打開Anaconda Prompt

3.添加清華鏡像

>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 --set show_channel_urls yes

4.添加Pytorch鏡像

>conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

注意:這里有個坑,必須是進(jìn)到win-64文件夾才可以!?。?/b>

5.安裝Pytorch(官網(wǎng):https://pytorch.org/)

>conda install pytorch torchvision cudatoolkit=10.0

注意:博主的cuda是10.0版本,根據(jù)不同的版本搜索自己的下載語句,記住去掉-c pytorch

-c pytorch參數(shù)指定了conda獲取pytorch的channel,在此指定為conda自帶的pytorch倉庫。我們安裝的時候去掉-c pytorch就可以使用清華鏡像快速安裝。

附:cuda版本查詢

>>>打開控制面板>>>點(diǎn)擊查看方式>>>選擇小圖標(biāo)

>>>點(diǎn)擊NVIDIA控制面板

>>>點(diǎn)擊左下角系統(tǒng)信息

>>>選擇組件>>>查看cuda版本

6.安裝完成

7.檢測是否安裝成功

>>>打開cmd>>>進(jìn)入python

輸入:>>>import torch?

輸入:>>>a = torch.Tensor([1.])?

輸入:>>>a.cuda()?

#正常則返回"tensor([ 1.], device='cuda:0')"

輸入:>>>from torch.backends import cudnn

輸入:>>>cudnn.is_acceptable(a.cuda())? ?

?#正常則返回 "True"

另:查詢pytorch版本

>>>print(torch.__version__)

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容