自己记录一下。以免以后忘记咋装的了。
1.anaconda和pycharm两个软件的安装,没啥说的。很多帖子都写的很完整了。
2.重点记录tensorflow-gpu的安装。
需要的东西:①cuda安装②cudnn安装③tensorflow-gpu安装
1)查看本机的cuda版本并下载
查看方式:NVIDIA控制面板--左下角系统信息--组件 里面即可查看到。
2)tensorfow社区查看所需要的tensorflow-gpu对应的cuda和cudnn版本
根据自己的cuda版本下载对应的cuda、cudnn。如果会*,则去NVIDIA官网下载,不能则搜索别人下载好的。
社区链接如下:从源代码构建 | TensorFlow (google.cn)https://tensorflow.google.cn/install/source#gpu
我的cuda是11.2,因此对应下载cudnn-8.1。
cuda安装时注意,选择自定义,且不要勾选visual studio和Driver。
3)anaconda中创建名为tensorflow的新环境:
anaconda prompt中输入:conda create --name tensorflow python=3.6(python版本根据需求自己选择)
激活tensorflow环境,anaconda prompt输入: activate tensorflow 发现括号中由base变为tensorflow
4)安装tensorflow
pip install --upgrade --ignore-installed tensorflow==1.15(版本自己选择)
验证是否安装成功:输入python进入代码环境,输入:import tensorflow as tf
无报错则成功:3.最后:这玩意就得多折腾,总共装了两天多才成功,期间出现很多很多问题,文中并未提到。还卸载了anaconda和pycharm一次。出现问题就网上搜,慢慢肯定能解决,动手能力就是折腾出来的。安装好后,终于跑起来一个基于tensorflow的代码了!
感谢csdn和知乎上的大佬们的帖子。
主要参考:tensorflow吐血安装—anaconda+tensorflow+pycharm安装(一) - 知乎 (zhihu.com)