CUDA安装以及基于pyenv和anaconda的TensorFlow和PyTorch安装与环境配置

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/huuuuuuuu/article/details/84109295

1.CUDA配置

Ubuntu

1.推荐16.04版本Ubuntu
2.NVIDIA显卡驱动安装（通过ppa，当前推荐375版本，通过ubuntu-drivers devices查看推荐的版本）

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
ubuntu-drivers devices
sudo apt install nvidia-375

或：

sudo add-apt-repository ppa:bumblebee/stable
sudo apt-get update
sudo apt-get install bumblebee bumblebee-nvidia

3.进入NVIDIA CUDA官网：https://developer.nvidia.com/cuda-downloads
4.依次选择：Linux -> x86_64 -> Ubuntu -> 16.04 -> runfile（local），点击下载run文件
5.将run文件置于~/Downloads文件夹下同时点击ctrl+alt+F1，进入terminal界面，然后依次输入账户密码登录
6.切换到~/Downloads文件夹，安装cuda，用户协议可以ctrl+c跳过

sudo sh cuda8.0.61375.26_linux.run

7.安装过程中，驱动安装选no，其余选yes或默认路径，如果安装出现问题，则将lightdm关闭后重试

sudo service lightdm stop

8.完成安装后按ctrl+alr+F7回到图形界面，如已关闭lightdm，则重启它

sudo service lightdm restart

9.将下面两行CUDA路径加入~/.bashrc,并执行下面命令

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

source ~/.bashrc

Windows

1.推荐使用Windows 10
2.NVIDIA官网下载安装显卡驱动（或跳过此步骤，直接安装CUDA内置的显卡驱动）
3.进入NVIDIA CUDA官网https://developer.nvidia.com/cuda-downloads 下载CUDA并安装

macOs

1.进入NVIDIA CUDA官网https://developer.nvidia.com/cuda-downloads 下载CUDA并安装

2.TensorFlow安装与配置

Ubuntu

1.安装 pyenv

curl -L https://raw.githubusercontent.com/yyuu/pyenv-installer/master/bin/pyenv-installer | bash

把下面语句追加入~/.bashrc文件种

export PATH="$HOME/.pyenv/bin:$PATH"
eval "$(pyenv init -)"
eval "$(pyenv virtualenv-init -)"

2.重启terminal，使用penv安装anaconda，当前推荐4.4.0版本

pyenv install anaconda3-4.4.0

安装后切换到anaconda环境

pyenv global anaconda3-4.4.0

然后用conda list查看当前所有的python库
3.安装TensorFlow 1.3版本(https://www.tensorflow.org/install/)
CPU版本：

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.3.0-cp36-cp36m-linux_x86_64.whl

GPU版本：

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp36-cp36m-linux_x86_64.whl

Windows

1.下载安装Anaconda 3.6版本：https://www.anaconda.com/download/#windows
2.打开Anaconda terminal，安装TensorFlow
CPU版本：

pip install --upgrade --ignore-installed tensorflow

GPU版本：

pip install --upgrade --ignore-installed tensorflow-gpu

macOS

可参照Ubuntu

3.PyTorch安装与配置

Ubuntu

1.安装完pyenv和anaconda后，参照官网https://pytorch.org/，直接
CPU版本：

conda install pytorch torchvision -c soumith

GPU版本：

conda install pytorch torchvision cuda80 -c soumith

Windows

PyTorch暂不支持Windows，可以用VMWare Player安装Ubuntu虚拟机

macOS

同Ubuntu：
CPU版本：

conda install pytorch torchvision -c soumith

GPU版本：
需要从源代码编译，见官网https://pytorch.org/