在虚拟环境中配置CUDA，cudnn, tensorflow-gpu

Handsome_zhou · 发表于 2022-1-25 13:10:06

马上注册，结交更多好友，享用更多功能^_^

您需要登录才可以下载或查看，没有账号？立即注册

x

本帖最后由 Handsome_zhou 于 2022-2-9 19:07 编辑

[i]failed to load the tensorflow runtime 一般是CUDA环境没配置好。[/i]

根据需要的tensorflow版本，选择对应的CUDA版本和cudnn版本。

激活虚拟环境：conda activate your_env

首先安装CUDA：conda install cudatoolkit=your_version

然后安装cudnn:conda install cudnn #不需要指定版本，cudnn会根据CUDA版本自动安装合适的版本

最后安装tensorflow:pip3 install tensorflow-gpu==1.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

配置完后就可以正常的使用了！

账号		自动登录	找回密码
密码			立即注册

[学习笔记] 在虚拟环境中配置CUDA，cudnn, tensorflow-gpu

马上注册，结交更多好友，享用更多功能^_^