1. 问题描述
external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:435]
执行命令之后,程序一直卡在这个地方,查完资料之后大家都说是CUDA
安装配置不正确
2. 重新配置CUDA环境
进入官方的github仓库Jax_installation寻找配置的方案:
红色方框中有一条pip
的安装命令,因为我的cuda
是11.4
何cudnn
也是8.2
,所以我选择执行这条命令。执行完成之后在执行下面红框中的的jax
安装命令:
3.测试结果
完成上述操作之后,执行命令,发现卡在那个问题了!