anaconda3.0+TensorFlow——GPU安装步骤
第一步、首先在anaconda官网下载3.0版本并安装(预装python的情况下)
anaconda下载链接:https://www.anaconda.com/products/individual#Downloads
第二步、安装cuda+cudnn
首先、搞清楚你的显卡是什么型号?自己的GPU算力是多少,支持什么版本号的CUDA?比如我的显卡型号是mx450相当于GTX1050m,但是找不到他的算力。这时候找比他第一档的比如GTX960m,于是我就用了GTX960m的配置,对应了CUDA10.1这个版本。
显卡与算力对照表:https://developer.nvidia.com/zh-cn/cuda-gpus根据你显卡的型号点进去就可以了。
不同CUDA版本对应的最小GPU运算能力和最低兼容驱动:参考链接
cuda下载链接:https://developer.nvidia.com/cuda-toolkit-archive
cudnn、TensorFlow与cuda的匹配表
cdunn下载链接:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
第三步、创建TensorFlow的环境并安装TensorFlow
首先、创建一个独立的环境
conda create -n tensorflow-gpu python=3.6.5(以3.6.5版本为例,实际情况要根据上面的表装对应版本的python环境)
然后激活你创建的独立环境:
conda activate tensorflow-gpu
最后安装TensorFlow-GPU
pip install --ignore-installed --upgrade tensorflow-gpu==1.7 #此处以版本号位1.7为例
这样就可以了!
可以根据自己的情况,将这个环境导入到pycharm中去,本文就不在赘述了。
借鉴于:https://blog.csdn.net/weixin_39228490/article/details/88851628