anaconda下安装pytorch
生活随笔
收集整理的這篇文章主要介紹了
anaconda下安装pytorch
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
- 一、安裝anaconda
- 二、安裝pytorch
- 1.打開Anaconda Prompt命令行
- 2. 使用清華鏡像
- 3.建立環境
- 4 安裝pytorch
- 5. 驗證是否安裝成功
一、安裝anaconda
官網一般顯示的是最新版本:
https://www.anaconda.com/products/individual
如果想要下載以往版本,可以在下面的網頁:
https://repo.anaconda.com/archive/
二、安裝pytorch
1.打開Anaconda Prompt命令行
在命令行格式下,輸入代碼,完成調用清華鏡像、建立pytorch環境、安裝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.建立環境
建立一個名字為pytorch_z的環境
代碼如下(示例):
彈出提示之后輸入y,即可安裝
之后輸入以下命令,即可看到已經安裝好了pytorch_z
conda env list
之后切換到自己新建的環境中
4 安裝pytorch
根據自己的cuda版本,在pytorch官網尋找代碼
輸入nvcc -V(注意大寫),即可看到cuda的版本
nvcc -V1)網站: pytorch官網
這里顯示只有10.2的,但是我是10.1,所以可以點擊
install previous versions of PyTorch
來找到歷史版本
這里可以看到我10.1版本的安裝命令
5. 驗證是否安裝成功
(1)在命令行左邊為 pytorch 環境中,輸入 python
(2)之后,輸入 import torch,如果沒有報錯,意味著 PyTorch 已經順利安裝了
(3)接下來,輸入 torch.cuda.is_available,如果是 True,意味著你可以使用 GPU,如果是 False,意味著只能使用CPU。
總結
以上是生活随笔為你收集整理的anaconda下安装pytorch的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Anaconda 查找安装的环境,以及移
- 下一篇: HJ9 提取不重复的整数