Open xmu-xiaoma666 opened 7 months ago
I find cuda affects performance significantly.
The cuda is 12.0, the performance on mme is as follow (very low):
Then cuda is 12.2, the performance on mme is as follow:
If I only have a machine with cuda 12.0, and I can't upgrade the Cuda version due to gpu driver. How can I reproduce the performance?
Hi @xmu-xiaoma666 , maybe you can try using torch version 2.0.1 and cuda 11.7 version which is used in llava1.5
I find cuda affects performance significantly.
The cuda is 12.0, the performance on mme is as follow (very low):
Then cuda is 12.2, the performance on mme is as follow:
If I only have a machine with cuda 12.0, and I can't upgrade the Cuda version due to gpu driver. How can I reproduce the performance?