解决方案:
由于电脑是笔记本,双显卡尝试过好多次总是在安装完CUDA后重启黑屏或是卡在登陆界面无法进入系统。应该是安装完cuda后负责图形界面的Intel不干活了。
以下是我配置平台的全过程,最终目的是使用tensorflow或theano并利用gpu加速。
1.首先安装Ubuntu16.04,之前装的是14.04,但总是装完CUDA后重启黑屏。在16.04的软件更新中可以同时更新到NVIDIA的独显和Intel的集显。
2.在系统设置的软件和更新中,安装好NVIDIA和Intel的专有驱动。
3.安装CUDA toolkit
sudo apt-get update
sudo apt-get install nvidia-cuda-toolkit
默认安装cuda 7.5.18
安装之后,是没有/usr/local/cuda*这个文件夹,也没有sample的
4.由于Ubuntu16.04的gcc和g++都是5.0版的,不兼容CUDA7.5版本,需要降级
sudo apt-get gcc-4.9 g++-4.9
cd /usr/bin
sudo rm gcc
sudo rm g++
sudo ln -s gcc-4.9 gcc
sudo ln -s g++-4.9 g++
5.安装cuDnn
下载cudnn 4.0 for cuda7.0 需要nvidia的开发者帐号登录
解压
tar -zxf cudnn-7.0-linux-x64-v4.0-prod.tgz
cd cuda
复制头文件到/usr/local/include
sudo cp include/cudnn.h /usr/local/include/
复制lib文件到/usr/local/lib
sudo cp lib64/* /usr/local/lib/
并编辑~/.bashrc 添加环境变量
export LD_LIBRARY_PATH=/usr/local/lib
6.安装theano
sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git
sudo pip install Theano
编辑配置文件
sudo gedit ~/.theanorc
加入
[global]
device = gpu
floatX = float32
[nvcc]
flags=-D_FORCE_INLINES
测试
7.安装tensorflow
sudo apt-get install python-pip python-dev # Ubuntu/Linux 64-bit,GPU enabled,Python 2.7 # Requires CUDA toolkit 7.5 and CuDNN v4. For other versions,see "Install from sources" below. export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp27-none-linux_x86_64.whl # Python 2
sudo pip install --upgrade $TF_BINARY_URL
测试
8.安装OpenAI Gym
因为后续可能会做一些增强学习的小项目,再此安装OpenAI Gym
git clone href=//github.com/openai/gym
cd gym
pip install -e . # minimal install
pip install -e .[all] # all install
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。