在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__)
