當前位置:網站首頁>使用Anaconda安裝Pytorch

使用Anaconda安裝Pytorch

2022-01-27 05:55:36 風吹我亦散

安裝Pytorch

首先安裝Anaconda, 一般下載最新版本可能會有問題, 進入Anaconda曆史版本網站 https://repo.anaconda.com/archive/
我下載的版本是:Anaconda3-5.1.0-Windows-x86_64.exe,可以複制一下然後在網頁上Ctrl+F直接搜就行了。

安裝的話直接運行exe程序即可,
需要注意的一點是,
在這裏插入圖片描述
此處點擊跳過。


在這裏插入圖片描述
點擊anaconda prompt


在這裏插入圖片描述
出現(base) ,說明安裝成功

查看環境conda info --env 或者 conda env list

創建環境conda create -n newName python=3.7
創建一個新的名為“newName”的虛擬環境,同時使用的python版本為3.7

切換環境conda activate newName

列出當前環境下有哪些包:pip list

PyTorch 安裝

PyTorch首頁:https://pytorch.org/

選擇下載版本的話先查看GPU型號:打開設備管理器,在顯示適配器處查看GPU型號
在這裏插入圖片描述
當電腦上有NVIDIA顯卡時,CUDA推薦選擇10.2(一般高版本都有一些問題,可能不好解决)

在這裏插入圖片描述
相關配置選擇好之後,給出了安裝指令,將安裝指令複制下來:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch


以防萬一,在桌面空白處右鍵點擊,選擇NVIDIA控制面板
在這裏插入圖片描述
查看版本是否 > 396.26在這裏插入圖片描述
這裏符合條件,可以安裝。


打開Anaconda的命令提示行(anaconda prompt
將剛剛複制的命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
粘貼上去,然後回車

如果失敗的話,可以嘗試用換源的方法解决,但是更推薦用下面這種方法
https://blog.csdn.net/weixin_45468845/article/details/122516155

就安裝完成了

安裝Jupyter

執行下列語句,再pytorch環境下安裝Jupyter所需的包

conda activate pytorch
conda config --set remote_read_timeout_secs 1000.0
conda install nb_conda

請添加圖片描述


輸入jupyter notebook
在這裏插入圖片描述

進入如下界面,
請添加圖片描述

點擊右上角的 New,選擇 Python[conda env.pytorch]*
輸入import torch,點擊運行
再輸入torch.cuda.is_available()
輸出 True,錶示我們在自己創建的pytorch環境中安裝好了jupyter包
請添加圖片描述

版權聲明
本文為[風吹我亦散]所創,轉載請帶上原文鏈接,感謝
https://cht.chowdera.com/2022/01/202201270555356021.html

隨機推薦