在win10基礎上已經安裝好了Anaconda,繼續安裝Pytorch。因為采用原本的下載方法實在太慢,還容易出現問題,現采取Pytorch清華鏡像下載方法,親測有效快速!!!
1.已經安裝好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/
注意:這里有個坑,必須是進到win-64文件夾才可以!!!
5.安裝Pytorch(官網:https://pytorch.org/)
>conda install pytorch torchvision cudatoolkit=10.0
注意:博主的cuda是10.0版本,根據不同的版本搜索自己的下載語句,記住去掉-c pytorch
-c pytorch參數指定了conda獲取pytorch的channel,在此指定為conda自帶的pytorch倉庫。我們安裝的時候去掉-c pytorch就可以使用清華鏡像快速安裝。
附:cuda版本查詢
>>>打開控制面板>>>點擊查看方式>>>選擇小圖標
>>>點擊NVIDIA控制面板
>>>點擊左下角系統信息
>>>選擇組件>>>查看cuda版本
6.安裝完成
7.檢測是否安裝成功
>>>打開cmd>>>進入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__)