pytorch运行过程中GPU利用率到达100%是batch_size还是多线程原因

pytorch运行过程中GPU利用率到达100%
我通过降低多线程，程序可以运行
但是我想提高多线程，请问有什么解决方法

两者对cpu影响都很大
你可以试试多线程，很小的batchsize