Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺
由于需要安装cuda8.0,,所以需要卸载cuda7.5.
在/usr/local/cuda/bin 目录下,有cuda 自带的卸载工具uninstall_cuda_7.5.pl
cd /usr/local/cuda/bin
sudo ./uninstall_cuda_7.5.pl
剩下步骤步骤如下：
1.安装显卡驱动
- sudo apt-get update
- sudo apt-get upgrade
- sudo add-apt-repository ppa:graphics-drivers/ppa
- sudo apt-get update
- sudo apt-get install nvidia-367
- sudo apt-get install mesa-common-dev
- sudo apt-get install freeglut3-dev
重启一下，使驱动生效，并使用以下命令测试驱动是否安装成功。
- nvidia-smi
2.安装CUDA
官网链接，在这里下载好CUDA后，使用以下命令安装。
- sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-rc_9.0.103-1_amd64.deb
- sudo apt-key add /var/cuda-repo-9-0-local-rc/7fa2af80.pub
- sudo apt-get update
- sudo apt-get install cuda
修改环境变量并使其生效。
- export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
- export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- export CUDA_HOME=/usr/local/cuda
这时，显卡驱动可能会被覆盖，重启一下万无一失。
3.安装cuDNN
官网链接，在这里下载好cuDNN，使用以下命令安装。
- tar -zxvf cudnn-9.0-linux-x64-v7.tgz
- sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
- sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
- sudo chmod a+r /usr/local/cuda/include/cudnn.h
- sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4.安装tensorflow
先激活gymlab
- source activate gymlab
安装python3-pip和python3-dev
- sudo apt-get install python3-pip python3-dev
由于我想安装最新版本的支持python3.5的tensorflow1.6.0，然后我尝试使用以下命令进行安装。
- sudo pip3 install –upgrade https://storage.proxy.ustclug.org/tensorflow/linux/gpu/tensorflow-1.6.0-cp35-cp35m-linux_x86_64.whl
报错了，404 not found，说明没有这个源，所以我又使用了以下命令进行安装。
- pip3 install tensorflow-gpu
这次没报错，而且下载的正是最新版的1.6.0，也正是支持python3.5的。难不成这条命令会根据本机的python版本来下载最新版本的tensorflow？这个问题待定。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/196884.html原文链接：https://javaforall.net
