对于个人而言，还是比较习惯使用 tensorflow 的 keras

安装 cuda 和 cudnn

安装方法可以见 pytorch 的安装，对于 tensorflow 已有的 gpu 支持可以见此。不提前装 cuda 和 cudnn 的方法见使用 conda 安装 cuda 和 cudnn

安装 tensorflow-gpu

由于我刚好符合 tensorflow-gpu 2.4.0，而 conda 还没有这个包，因此使用 pip 进行安装

1	pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

镜像源可能下载速度还是不行，可以直接本地安装

验证 gpu

运行代码

import tensorflow as tf

print(tf.__version__)
print(tf.config.list_physical_devices())

输出结果如下

2021-03-26 17:00:35.277510: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudart64_110.dll2.4.0
2021-03-26 17:00:39.259583: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2021-03-26 17:00:39.261539: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library nvcuda.dll      
2021-03-26 17:00:39.307535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:01:00.0 name: GeForce GTX 1050 computeCapability: 6.1
coreClock: 1.493GHz coreCount: 5 deviceMemorySize: 2.00GiB deviceMemoryBandwidth: 104.43GiB/s
2021-03-26 17:00:39.308404: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudart64_110.dll2021-03-26 17:00:39.322068: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublas64_11.dll 
2021-03-26 17:00:39.322414: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublasLt64_11.dll
2021-03-26 17:00:39.329903: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cufft64_10.dll  
2021-03-26 17:00:39.332538: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library curand64_10.dll 
2021-03-26 17:00:39.347776: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusolver64_10.dll
2021-03-26 17:00:39.357722: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusparse64_11.dll
2021-03-26 17:00:39.362238: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudnn64_8.dll   
2021-03-26 17:00:39.362701: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU'), PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]