【教程】基於Ubuntu系統的PyTorch虛擬環境配置

目錄

• 一、PyTorch虛擬環境配置
• 二、PyTorch虛擬環境使用
• 三、常用命令

• Editor: Veagau
• Time: 2019/10/17

一、PyTorch虛擬環境配置

• 該部分操作均在終端（terminal）命令行中執行，在桌面右擊滑鼠新建terminal即可。

• 若系統已經安裝Anaconda，則可以跳過第一步（檢驗方法：命令行執行conda list，有返回結果則表示已經安裝了Anaconda）。

1. 安裝Anaconda。進入到安裝包所在文件夾，執行：

   bash Anconda3-x.x.x-Linux-x86_64.sh

2. 創建虛擬環境。

   conda create -n <your-virtualenv-name> python=3.6

3. 激活虛擬環境。

   source activate <your-virtualenv-name>

   如果激活成功，則命令指示符首部將出現由小括弧包含的虛擬環境名稱。

4. 安裝PyTorch。

   • 此處PyTorch相關組件包的安裝是在虛擬環境中進行的，與系統環境隔離。
   • 打開PyTorch官網，根據機器配置選擇參數，複製網站提供的命令執行即可。
   • 由於PyTorch從國外伺服器下載過於緩慢，所以建議首先更換conda鏡像下載源，以節省時間。
   • 一般的深度學習機型的顯卡為RTX系列的，在選擇參數時CUDA項建議選擇10.1版本，此處採用conda（包管理工具，類似與pip）進行安裝，所以Package項建議選擇Conda。

   1. 更換國內conda鏡像源。逐行執行下列命令：

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

   2. 安裝PyTorch。執行從PyTorch網站上複製的命令：

      conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

      此條命令實際上同時安裝了pytorch、torchvision、cudatoolkit三個包，執行過程中會檢測依賴的其他包，自動進行安裝。

5. 檢測PyTorch環境。

   1. 在終端命令行執行python命令，進入Python環境。

   2. 導入PyTorch包，並查看版本號

        import torch as t #在Python環境下PyTorch的包名為torch
        print(t.__version__) #d列印PyTorch包的版本信息

      如返回1.x.x等數字，則表明Pytorch框架安裝成功。

二、PyTorch虛擬環境使用

針對PyCharm這一集成開發環境（IDE），以下為使項目運行於創建的虛擬環境的具體操作。

2.1 新建項目

1. File→New Project

2. 項目信息設定

   1. 指定Location和項目名稱；

   2. 選擇Project Interpreter即項目編譯器

      1. 選定Existing interpret

      2. 點擊Interpret框右側的選擇按鈕

      3. 選擇Conda Environment

      4. 點擊Interpret框右側的選擇按鈕

      5. 前往Anaconda安裝目錄找到虛擬環境中的Python程式。

         路徑一般為：/home/user/Anaconda3/envs/your_virtualenv_name/bin/python

      6. 點擊ok→create即可

3. 檢查虛擬環境是否啟用成功。

   新項目創建後需要花一段時間進行Python編譯器更新以及依賴包的導入，請耐心等待。

   1. 點擊視窗下方的Terminal選項卡，打開終端。
   2. 若看到命令指示符首部出現由小括弧包含的虛擬環境名稱 ，則表明虛擬環境啟用成功。

2.2 歷史項目

1. File→Settings→Project：your_project_name→Project Interpret
2. 點擊Project Interpret框→Show All
3. 點擊➕（加號）按鈕，添加虛擬環境中的Python編譯器
4. 剩餘步驟同新建項目的2.2.3之後的操作。

三、常用命令

3.1 虛擬環境相關

• 查看

  conda list env

• 創建

  conda create -n virtualenv_name python=3.x` #x指代python版本號

• 激活

  source activate <virtualenv_name>

• 退出

  deactivate

• 克隆

  conda create -n <new_venv> --clone <old_venv>

• 刪除

  conda remove -n <venv_name> --all

3.2 工具包相關

• 查看已安裝工具包

  pip list

• 查看具體工具包信息

  pip show <包名>

• 安裝工具包

  pip install <包名>

• 安裝指定版本的工具包

  pip install <包名>==<版本號>

• 卸載工具包

  pip uninstall <包名>