防坑 centos7 安装 CUDA9.0 + cudnn7.1 +TensorFlow GPU版1.6.0/1.8.0
为了加快模型的训练和提高效率,于是乎入手TensorFlow的GPU版!
一、安装CUDA
下载地址:https://developer.nvidia.com/cuda-90-download-archive
首先检查自己的电脑有没有NVIDIA的显卡
检测环境:传输门 CUDA官方安装说明https://docs.nvidia.com/cuda/cuda-installation...
原创
2018-06-16 15:46:02 ·
7144 阅读 ·
1 评论