Open Faker0715 opened 2 years ago
我试了torch1.6和1.8都报这个错误
我也遇到了
同样遇到了,请问这个问题怎么解决呀
请问您的这个问题解决了吗?是torch版本的问题吗?
我用的torch1.11也报这个错
我也遇到了,请问是怎么解决的
把这块注释掉就可以了,这一块应该打印出的是gpu的使用情况 train.py 的242 243 244行
# gpu_mem = cudause.gpu_mem_get(_id=0)
# record.log_single(key="gpu_mem", value=gpu_mem)