Anaconda安裝Pytorch鏡像詳細教程

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

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容