问题
- 通过控制台直接输出，可以正常输入调用tensorflow gpu
- 测试代码：
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
- 测试代码：
- 使用pycharm启动之后，错误提示 无法找到gpu了
解决
- 使用pycharm启动的时候，需要添加环境变量，不然无法识别cuda
- 编辑启动配置，添加环境变量：
LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64cuda需要修改为实际的安装位置
文章评论