CUDA11.7配置tensorflow-gpu

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
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值