windows10+anaconda3+python3.7+pycharm+gpu–tensorflow 2.1
NVIDIA GeForce GTX 1650版本441.22安装的CUDA版本是10.2.9
CUDA
有关显卡,显卡驱动,cuda ,cudnn是什么,以及之间的关系可自行搜索
在左下角搜索框搜索NVIDIA进入控制面板,查询当前版本
算力需大于3.1
查询CUDA版本和显卡驱动版本 需按对应下载
CUDA下载 下载非常慢 😢,安装就是一直无脑下一步
测试win+R: nvcc -V
出现上图证明成功。
cuDNN
cuDNN下载 直接下载要注册,右键选择复制链接,在打开迅雷即可,或者直接右键迅雷下载,我选择的是for cuda9.0的版本,没选择最新的,好像问题不大,也有可能比较慢,
最后得到一个zip文件,解压后得到三个文件夹,如下所示:
将bin ,include 和lib文件夹的内容分别复制到cuda安装路径下与cuda的bin ,include 和lib文件夹合并。
测试安装是否成功
win+R :cmd
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
.\deviceQuery.exe显示如下结果说明安装成功:
Tensorflow
win+R
python -m pip install --upgrade pip
pip install tensorflow-gpu==2.1
首先更新pip,第二句容易下载超时,不行则换成下面这句
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
指定镜像源会比较快,安装需要一点时间
pycharm
新建一个项目文件,将anaconda和pycharm绑定
file->settings->project interpreter ->右边小齿轮->show all ->+号->一般在c盘user路径下选择anaconda文件中的Python.exe
新建一个python程序
import tensorflow as tf
tf.__version__
首次运行出现如下错误
Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found
Ignore above cudart dlerror if you do not have a GPU set up on your machine.
先来排查错误,显示“cudart64_101.dll not found”,那是因为我安装的cuda为10.0版本的,“cudart64_101.dll”是cuda10.1这个版本中才含有的(cuda10.2貌似也没有)。
因此我们需要手动将“cudart64_101.dll”添加到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin”中去。
再次运行成功
希望对你有所帮助!