- 博客(4)
- 收藏
- 关注
RSS订阅原创 Resnet代码运行时显示GPU内存不足
此问题常见于较多层的训练模型，此时需要改变batch-size的值，但需要注意的是batch-size的值仍需满足2的几次幂这一条件。可见结果与预想的大致相同。resnet18训练结果。resnet50训练结果。
2023-08-03 22:02:06
361
原创 pycharm中使用pytorch的一些问题
由于使用了Anoconda的虚拟环境创建了两个环境，一个名为xypytorch，一个名为pytorch1，在Anaconda prompt内使用pip install tensorboard命令安装tensorboard时，虽然包被导入到pytorch1环境内，但是tensorboard.exe文件并没有进入pytorch1包对应的Sciripts文件夹下，而是进入了另一个虚拟环境xypytorch的Scripts文件夹下。找到tensorboard.exe文件的目录，又找到了问题。
2023-06-06 09:18:00
205
原创 深度学习在pycharm中python解释器配置
如图所示，在配置时一定要在所创建的虚拟环境里面找到python包，不能直接在Anaconda的安装位置里面找python包。如果一开始在解释器选项里面找不到python包，需要重启电脑来重新配置。
2023-06-01 16:39:29
252
原创 RTX4050的cuda版本确定。
我的Driver版本是12.1，所以 CUDA Runtime应小于12.1，对于4050显卡来说，已经查到 CUDA Runtime版本可以使用11.5--11.8（对于40系显卡来说目前都是这样），所以下载11.8版本，总之不论如何，一定要有 CUDA Driver 版本 >= CUDA Runtime 版本。确保自己的 CUDA Driver 版本 >= CUDA Runtime 版本。
2023-05-31 08:54:34
2868
5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人