Ubuntu16.04 GPU环境配置tensorflow
Ubuntu16.04 GPU环境配置tensorflow
查看电脑是否安装GPU显卡驱动
- 可通过桌面右上角设置,关于这台电脑查看,是否有GPU显卡驱动
2.若没有显卡驱动,可参考以下方法,安装显卡驱动
https://blog.csdn.net/weixin_42270433/article/details/96476178
3.再次查看,可看到图形那里,已经显示出显卡的型号。
4.在终端输入命令:nvidia-smi,可看到安装的显卡驱动型号
更新python版本和pip版本
更新python版本
系统默认Python版本为2.7,需要更改python版本,终端命令操作如下:
$cd /usr/bin
$sudo rm -rf python
$sudo ln -s /usr/bin/python3 /usr/bin/python
更新pip版本
sudo apt-get install python3-pip
发现无法安装pip3,解决办法:
sudo apt-get --fix-broken install
pip install -U pip
sudo apt-get install python3-pip
pip install --upgrade pip
安装Cuda9.0
1.Cuda下载地址:
https://developer.nvidia.com/cuda-toolkit-archive
2.Cuda安装命令:
1.$sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64.deb
2.