- 博客(4)
- 收藏
- 关注
原创 Resnet代码运行时显示GPU内存不足
此问题常见于较多层的训练模型,此时需要改变batch-size的值,但需要注意的是batch-size的值仍需满足2的几次幂这一条件。可见结果与预想的大致相同。resnet18训练结果。resnet50训练结果。
2023-08-03 22:02:06
533
原创 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
262
原创 深度学习在pycharm中python解释器配置
如图所示,在配置时一定要在所创建的虚拟环境里面找到python包,不能直接在Anaconda的安装位置里面找python包。如果一开始在解释器选项里面找不到python包,需要重启电脑来重新配置。
2023-06-01 16:39:29
322
原创 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
4407
5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人