在虚拟环境中配置CUDA，cudnn, tensorflow-gpu
本帖最后由 Handsome_zhou 于 2022-2-9 19:07 编辑failed to load the tensorflow runtime 一般是CUDA环境没配置好。
根据需要的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
配置完后就可以正常的使用了！
页:
[1]