Win10+RTX3080+TensorFlow2.4.0

一、环境匹配

RTX3080要求CUDA版本>=11.0。这是一个很棘手的限制。我们查看Tensorflow官方已支持版本（2021年4月27日）

我们要使用RTX3080炼丹，只能使用tensorflow_gpu-2.4.0+CUDA11.0+cuDNN8.0这样的组合，其中tensorflow_gpu和CUDA版本要求很严格，而cuDNN版本相对没有那么严格。

二、具体实践

为了省事，我们使用anaconda3。
1、创建环境

conda create -n tf2 python=3.8

2、激活环境

conda activate tf2

3、安装cudatoolkit11.0

conda install cudatoolkit=11.0

4、安装tensorflow_gpu-2.4.0

pip install tensorflow-gpu==2.4.0

5、修改dll文件名

我们直接调用gpu运算会提示无法加载cupti64_110.dll文件，这会导致程序仍用cpu进行计算，报错信息如下：

2021-04-27 21:15:09.044047: I tensorflow/core/profiler/lib/profiler_session.cc:136] Profiler session initializing.
2021-04-27 21:15:09.044152: I tensorflow/core/profiler/lib/profiler_session.cc:155] Profiler session started.
2021-04-27 21:15:09.044265: I tensorflow/core/profiler/internal/gpu/cupti_tracer.cc:1365] Profiler found 1 GPUs
2021-04-27 21:15:09.045446: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cupti64_110.dll'; dlerror: cupti64_110.dll not found
2021-04-27 21:15:09.046315: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cupti.dll'; dlerror: cupti.dll not found

我们只需进入anaconda安装的虚拟环境路径下寻找即可，我的虚拟环境名为tf2，因此路径为：anaconda3\envs\tf2\Library\bin，找到
cupti64_2020.1.1.dll文件，并将其文件名修改为cupti64_110.dll即可！

至此，大功告成！

参考：
[1]:https://tensorflow.google.cn/install/source_windows#gpu
[2]:https://blog.csdn.net/weixin_43233764/article/details/111536819