文章目錄

• 一、安裝anaconda
• 二、安裝pytorch
• • 1.打開Anaconda Prompt命令行
  • 2. 使用清華鏡像
  • 3.建立環(huán)境
  • 4 安裝pytorch
• 5. 驗(yàn)證是否安裝成功

---

一、安裝anaconda

官網(wǎng)一般顯示的是最新版本：

https://www.anaconda.com/products/individual

如果想要下載以往版本，可以在下面的網(wǎng)頁：

https://repo.anaconda.com/archive/

二、安裝pytorch

1.打開Anaconda Prompt命令行

在命令行格式下，輸入代碼，完成調(diào)用清華鏡像、建立pytorch環(huán)境、安裝pytorch、測試pytorch過程

2. 使用清華鏡像

分別輸入以下四個代碼

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 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

3.建立環(huán)境

建立一個名字為pytorch_z的環(huán)境
代碼如下（示例）：

conda create -n pytorch_z python=3.7

彈出提示之后輸入y，即可安裝

之后輸入以下命令，即可看到已經(jīng)安裝好了pytorch_z

conda env list

之后切換到自己新建的環(huán)境中

conda activate pytorch_z

4 安裝pytorch

根據(jù)自己的cuda版本，在pytorch官網(wǎng)尋找代碼

輸入nvcc -V（注意大寫），即可看到cuda的版本

nvcc -V

1）網(wǎng)站: pytorch官網(wǎng)

這里顯示只有10.2的，但是我是10.1，所以可以點(diǎn)擊
install previous versions of PyTorch
來找到歷史版本

這里可以看到我10.1版本的安裝命令

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

5. 驗(yàn)證是否安裝成功

（1）在命令行左邊為 pytorch 環(huán)境中，輸入 python

（2）之后，輸入 import torch，如果沒有報(bào)錯，意味著 PyTorch 已經(jīng)順利安裝了

（3）接下來，輸入 torch.cuda.is_available，如果是 True，意味著你可以使用 GPU，如果是 False，意味著只能使用CPU。

總結(jié)

以上是生活随笔為你收集整理的anaconda下安装pytorch的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。