- 博客(2)
- 收藏
- 关注
原创 安装pytorch_lightning要点
不记得是哪一个了,重要的是后面跟版本号)(安装完了之后记得重新打印一下torch版本看一下,有可能安装lightning时把原先的GPU版torch卸载,安装了CPU版的torch,这也是坑所在,注意注意!没出现报错就是安装成功了(或者直接运行下面的最简LigntningModule)torch和lightning都使用pip安装。(0.6.0其实是瞎碰的,然后报错竟然没有了)以上为个人经验记录,仅供参考。(折腾了一上午,简直崩溃!至此,应该是安装成功了。
2023-05-29 14:46:50 3489 2
原创 报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.
中看到了两种解决办法,但是由于在 TensorFlow 2 中,tf.GPUOptions 和 tf.ConfigProto 已经被弃用,chatGPT推荐使用 tf.config.experimental 模块中的 API 进行 GPU 配置。ChatGPT一直和我说是CUDA和cuDNN的版本问题,但是上网搜索了TensorFlow与之对应的版本,发现没有问题。
2023-04-11 11:23:02 660 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人