CUDA11.7配置tensorflow-gpu:
先上官网的图 https://tensorflow.google.cn/install/source_windows?hl=zh-cn#gpu
不用纠结CUDA版本必须严格对应,选取对应的python版本和tensorflow-gpu-2.6.0即可。
步骤:
# 配置conda环境
conda create -n tf python=3.7
conda activate tf
pip install tensorflow-gpu==2.6.0
# 测试
import tensorflow as tf
tf.test.is_gpu_available() #旧版,返回True表示成功
tf.config.list_physical_devices('GPU') #返回GPU列表表示成功
可能的报错:
1.Could not load dynamic library ‘libcudart.so.11.0’
conda install cudatoolkit=11.0
2.Could not load dynamic library ‘libcusolver.so.11’
找到conda环境lib文件,发现下载的是libcusolver.so.10而不是11
mv libcusolver.so.10 libcusolver.so.11
3.Could not load dynamic library ‘libcudnn.so.8’
conda install -c anaconda cudnn