- 博客(4)
- 收藏
- 关注
原创 pycharm删除一个工程的配置文件
在pycharm创建一个工程后会自动在工程下面生成.idea的文件夹。.idea包含了环境变量、编译器版本等工程信息,删除该文件夹,即可清除这些配置信息。
2021-03-12 10:22:29
4112
原创 pytorch的torch.cuda.set_per_process_memory_fraction()函数介绍
函数功能:在pytorch中设置显存使用比例,即能够完成显存的使用上限设置。torch.cuda.set_per_process_memory_fraction(0.5, 0) 参数1:fraction 限制的上限比例,如0.5 就是总GPU显存的一半,可以是0~1的任意float大小; 参数2:device 设备号; 如0 表示GPU卡 0号;功能解释如下:Set memory fraction for a process. The fraction is used to l
2021-03-11 15:08:49
6133
1
原创 PyTorch上进行GPU显存限制/切分的函数
PyTorch上进行GPU显存限制/切分的函数形态:torch.cuda.set_per_process_memory_fraction(0.5, 0)
2021-03-08 21:31:29
2273
原创 PyTorch上限制GPU显存的函数
torch.cuda.set_per_process_memory_fraction(0.5, 0) 参数1:fraction 限制的上限比例,如0.5 就是总GPU显存的一半,可以是0~1的任意float大小; 参数2:device 设备号; 如0 表示GPU卡 0号;
2021-03-08 15:59:56
6359
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人