Ubuntu和Centos安装Tensorflow教程&PyTorch

更多请关注：

• 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盘时有两点需要特别注意：

1. U盘接口最好需要是2.0，对于3.0的接口有的电脑主板不一定原生支持

2. 对于新安装硬盘（例如新买电脑），尤其是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

<