基于pyenv和anaconda的TensorFlow和PyTorch安装与环境配置
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/