更多请关注:
- https://github.com/chanhal
- https://www.zhihu.com/people/chanhal
- https://github.com/chanhal/config_ml_env (原版PDF下载)
- chanhal@outlook.com
文字版
Ubuntu和Centos安装Tensorflow教程&TyTorch
Ubuntu安装TensorFlow
安装Ubuntu 16.04 LTS
下载地址:http://releases.ubuntu.com/16.04/
[外链图片转存失败(img-SQjaHmqR-1568858900081)(media/3e7357d001043b9d534ae2d8760315d2.png)]
Ubuntu
ISO下载完成之后利用ultraiso写入U盘,制作系统启动盘,将系统设置为U盘启动,机器重启后就开始安装系统(DELL台式机,F12可以设置U盘启动)。制作启动U盘时有两点需要特别注意:
-
U盘接口最好需要是2.0,对于3.0的接口有的电脑主板不一定原生支持
-
对于新安装硬盘(例如新买电脑),尤其是SSD,安装前一定要全盘格式化一遍,不然安装过程中会出现无法分区,显示该盘的空间也会异常
安装CUDA toolkit 9.0和cuDNN 7.1
参考:
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html(推荐)
[外链图片转存失败(img-fsWLqjEp-1568858900082)(media/939af5add09a4538dfdd59f688ba6f75.png)]
安装CUDA Toolkit 9.0
- 通过下面命令产看机器的GPU硬件型号,然后在http://developer.nvidia.com/cuda-gpus这里查看机器的GPU是否支持CUDA。
$ lspci | grep -i nvidia
-
<