danfenghong / IEEE_TPAMI_SpectralGPT

Hong, D., Zhang, B., Li, X., Li, Y., Li, C., Yao, J., Yokoya, N., Li, H., Ghamisi, P., Jia, X., Plaza, A. and Gamba, P., Benediktsson, J., Chanussot, J. (2024). SpectralGPT: Spectral remote sensing foundation model. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2024. DOI:10.1109/TPAMI.2024.3362475.

191 stars 20 forks source link

Standardise number of GPU Used? #11

Open robmarkcole opened 7 months ago

robmarkcole commented 7 months ago

From the paper 4 NVIDIA GeForce RTX 4090 GPUs, are used. However for Eurosat script --nproc_per_node=2 - I have a machine with 4GPU but setting --nproc_per_node=4 resulted in an error (RuntimeError: CUDA error: invalid device ordinal). I propose to standardise on 4 GPU, or documenting how to adjust the GPU usage

ArizaWilmerUTAS commented 2 months ago

I also join to standarise the gpu # as with one single gpu I have to remove in main_finetune. '''

os.environ["CUDA_VISIBLE_DEVICES"] = "2,3,4,5"

'''

Without this line the code runs with 1 GPU