赞
踩
Anaconda官网(https://www.anaconda.com/)直接下载一个Anaconda,安装。
我这里用的python3.8,cuda 12.0以及对应的cudnn版本8.9
先激活前面创建好的虚拟环境:conda activate tensor_env
查看cuda12对应的GPU版本,官网没看到,网上搜了下,可以安装tensorflow-gpu2.7.0,对应的keras版本也是2.7.0,原始安装会有一个protobuf,但是版本不对应,需要重装下
pip install tensorflow-gpu2.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install keras2.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install protobuf==3.19.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装成功后,打开jupyterlab或者vscode,执行以下命令,能看到GPU说明能正常使用了
import tensorflow as tf
print(tf.config.list_physical_devices(“GPU”))
print(tf.config.list_physical_devices(“CPU”))
输出:
[PhysicalDevice(name=‘/physical_device:GPU:0’, device_type=‘GPU’)]
[PhysicalDevice(name=‘/physical_device:CPU:0’, device_type=‘CPU’)]
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。