PyTorch 最新安裝教程


PyTorch 最新安裝教程

1. 安裝 Anaconda
Anaconda 是一個用於科學計算的 Python 發行版,支持 Linux, Mac, Windows, 包含了眾多流行的科學計算、數據分析的 Python 包。

Anaconda 安裝包可以到 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下載 or 也可以去官網下載。

2. 檢查顯卡,更新驅動
建議是首先更新驅動,全都按最新的東西來安裝
https://www.nvidia.cn/Download/index.aspx?lang=cn#

在這里選擇自己的顯卡型號,下載安裝即可,下載很快也很簡單,建議大多數人都更新一下顯卡驅動,避免不必要的麻煩。

 

 

  1. 安裝好后進入NVIDIA控制面板,鼠標右鍵可進入

 

 點擊組件找到自己CUDA的版本號

 

 或者 win+R—> cmd ,進入命令行,輸入:

nvidia-smi

 

 

如果沒有這條命令,則需要添加環境變量(百度配置)一般電腦都可以直接用,我的顯卡是:GF:MX250

也可獲得版本號:

 

 

備注:第二步不一定是必須的,如果你的電腦較新,那么不更新也可以直接進行下面操作,我的電腦是19年買的,中間好像也沒更新驅動,這一步直接跳過,最后也安裝成功了。

3. 創建PyTorch環境
不同的項目需要不同的虛擬環境,可以處理不同版本的項目之間不兼容問題。

  1.進入 Anaconda prompt 命令窗口

 

 2. 輸入以下內容:

conda create -n PyTorch python=3.8

PyTorch是虛擬環境名字(可以隨意設置),3.8 是python版本,都可以按自己需求改,一定要指定具體 python 版本。

 

 然后按 y繼續安裝所需的各種依賴包。

  1. 創建成功后,輸入以下命令:
conda info --envs

可以看見自己的所有環境

 

 如果出現錯誤,可能是外網下載過慢,需要配置國內鏡像源。

  4. 配置清華TUNA鏡像源
TUNA 提供了 Anaconda 倉庫與第三方源(conda-forge、msys2、pytorch等,查看完整列表)的鏡像,各系統都可以通過修改用戶目錄下的 .condarc 文件。Windows 用戶無法直接創建名為 .condarc 的文件,可先執行 conda config --set show_channel_urls yes 生成該文件之后再修改。

注:由於更新過快難以同步,TUNA不同步pytorch-nightly, pytorch-nightly-cpu, ignite-nightly這三個包。
  1. 生成 .condarc 文件

 Anaconda prompt 命令窗口,中輸入:

conda config --set show_channel_urls yes

之后可以在 C:\Users\xxx 中看到 .condarc 文件

 

 2. 記事本打開 .condarc 文件,重寫其中的內容。

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

 

 

即可添加 Anaconda Python 免費倉庫。

Anaconda prompt 命令窗口運行 conda clean -i 清除索引緩存,保證用的是鏡像站提供的索引。

5. 安裝 PyTorch

  1. 進入pytorch官網:https://pytorch.org/

  2. 網頁下拉,即可看到下圖,官網會自動根據你的電腦,顯示的即是你可安裝的CUDA版本,並給出安裝命令。

 

 

  1. 復制官網提供的命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
  1. 打開 Anaconda prompt 命令窗口,進入你剛剛所創建的環境(我的命名是PyTorch)
conda activate PyTorch

進入環境

 

   5. 最后輸入官網提供的命令,即可下載

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 

 

 

 

每個人的命令,會由於電腦配置而各不同

注意:下載安裝過程,可能因為某些原因,並不是很順利,但遇到問題不要慌,另外網速一定要好,避免出現不必要的錯誤。

6. 測試

打開 Anaconda prompt 命令窗口,激活環境,輸入python,進入python開發環境中

import torch
torch.cuda.is_available()
True

 

 看到True的那一刻,我真的開心,終於成功了。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2026 CODEPRJ.COM