Ubuntu服務器上安裝pytorch環境

任務描述

Linux系統版本：Ubuntu16.04

cuda==9.2

conda == 4.8.0

python==3.6.4

pytorch==1.6.0

此過程是自己在window和linux安裝時環境記錄的過程后續整合的一個小tips，方便之后復盤

安裝步驟

• conda的因為服務器上已經裝好了，這里主要記錄一下環境的創建，包的安裝

  查看conda版本

   conda --version
  

  更新conda

   conda update conda
  

• 首先到CUDA官網找到9.2版本，我這里是直接下載run文件到本地，然后在上傳到服務器上，然后在服務器中使用sh命令直接運行這個run文件

  下載鏈接：

• 安裝好后，如果遇到CUDA多版本的問題，參考上一篇筆記

• 設置清華的鏡像源，原本的anaconda用的是外網的源，會導致下載速度特別慢和頻率很高的掉包

  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
  

• 然后按照如下修改.condarc文件

• 使用conda創建虛擬環境,我這里使用的pytorch版本是1.6.0，命令為py364-pt16，其中py364表示python版本是3.6.4，pt16表示pytorch版本是1.6.0

   conda create -n py364-pt16 python=3.6.4
  

  過程中會提示導入庫，直接y后回車即可

  查看環境是否創建成功

   conda info -e
  

  查看安裝環境中的包,-n限定查看虛擬環境

   conda list -n py364-pt16
  

  激活環境

   activate py364-pt16
  

  退出當前環境

   conda deactivate
  

• 安裝pythorch

  這里記錄一個當時遇到的問題，當時是直接使用官網的conda命令安裝的，安裝好后查看的pytorch版本不是1.6.0

   conda install pytorch torchvison cudatoolkit=9.2 pytorch
  

  后續使用pip又重新安裝了對應的版本

   pip install torch==1.6.0 torchvision==0.7.0
  

  查看torch版本

   pip list |grep pytorch
  

版本是相應版本，但是試着運行一段代碼遇到了pytorch版本1.6和cuda版本9.2不匹配的問題

解決辦法：先將pytorch、torchvision都卸載掉，使用官網上的pip重新安裝后

 pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

考慮下載速率的問題，我因為之前加了清華的鏡像源就想着不直接用-f后面鏈接下載，但是命令提示報錯，網友測試這里不用用國內的鏡像源

至此，環境的需要安裝的關鍵部分就已經成功了，根據這個問題總結一下：之后安裝一些庫首先還是官網相應的版本安裝命令提示，找的一些博客上的命令會根據使用情況不同缺少一些東西導致后期的一些錯誤

• 另外可能會根據程序的使用去安裝一些工具包，一般地，版本都沒有大要求，版本要是對應不上也會有相關的提示，通常按照以下命令可以安裝大部分需要的工具包

  pip install packagename(包名)