xiajunniat的博客

原创 centos GPU tensorflow pytorch 深度学习 环境搭建

简要记录一下深度学习开发环境的搭建步骤，懒得编辑了，将就着吧。总体流程参考文章：https://www.cnblogs.com/learn-the-hard-way/p/12318980.html按照这篇来做基本ok，要注意的是版本问题，因为cuda+cudnn+tensorflow+pytorch这些都要版本匹配的，我之前装的cuda10.2，cudnn和pytorch能找到匹配版本，但是tensorflow2.2.0是2020.5.8号发布的，国内还没有镜像支持，只能回退cuda到10.1，多花了不