轉載自：https://www.zhihu.com/people/an-xing-le-20/posts
算法開發環境搭建
一. CUDA 安裝
1. 驅動安裝
首先需要自己的GPU型號，這裏執行
lspci | grep -i vga
然後我們可以進入NVIDIA官網查看對應的驅動，並下載。
2. 下載CUDA
我們進入cuda toolkit 下載最新cuda開發包（請與 pytorch、tensorflow 版本相對應）。
3. 安裝
下載的時候，實際已經可以看到安裝方式了：
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.runsudo sh cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run
但是可能還是會遇到一些坑，多是一些權限等問題，推薦超級管理員安裝。
4. cuDNN安裝
進入cuDNN介紹目錄http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html。找到和CUDA對應的cudnn開發包。
// 切換到壓縮包所在位置，解壓
$ tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
// 複製 文件到cuda 目錄下
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
// 修改文件權限
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
二. Python環境安裝
1. 下載
首先需下載 Anaconda開發包，請登錄 Anaconda官網 或者執行
wget https://repo.anaconda.com/archive/Anaconda3-2019.07-Linux-x86_64.sh
來下載安裝二進制文件。
2. 安裝
本着linux用戶“誰使用，誰負責”的原則，用戶在本目錄下直接執行
sh Anaconda3-XXX.sh
然後一路默認 “yes” （可以不安裝Vscode）即可（注意：這裏需要仔細查看，眼鏡不要離開屏幕）。
3. 驗證使用
bash下輸入python，可看到python運行環境應該爲：
[外鏈圖片轉存失敗,源站可能有防盜鏈機制,建議將圖片保存下來直接上傳(img-0LyIMcAC-1588772568523)(http://10.30.1.6:33511/uploads/file_image/ueditor/php/upload/image/20190924/1569308459900916.png)]
具體版本號等視自己安裝情況而定。
4. 安裝虛擬環境
不同的項目往往需要安裝多個不同的python解釋器，不同的python開發包。 爲方便進行多python環境管理、開發（各個項目的python環境互不干擾），開發環境需要多套不同的python解釋器以及開發包與項目一一對應。這個時候python虛擬環境安裝管理就極爲便捷。
本着簡潔易用原則，我們使用conda來進行多python虛擬環境的管理。
// 安裝 版本爲X.X的python解釋器
conda create -n your_environment_name(具體是你的python虛擬環境名) python=X.X(這裏替換爲你的python版本，例： 2.7, 3.6, 3.7...)
// 激活虛擬環境（開始使用該環境, 項目開發，包管理....）
conda activate your_environment_name
5.修改文件
打開文件.bash_profile，添加如下
# cuda10.0
export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export LC_CTYPE ='zh_CN.utf8'
三. pytorch、tensorflow 環境安裝
這個如果不會，請直接回大學讀大一吧。地球已經拯救不了你了。
這裏注意兩點：更新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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安裝得時候不要加 -c參數
conda install pytorch torchvision cudatoolkit=10.0