search

LiyaoTang / contrastBoundary

Contrastive Boundary Learning for Point Cloud Segmentation (CVPR2022)
MIT License
139 stars 11 forks source link

How to set the GPU device? #7

Closed whuhxb closed 2 years ago

whuhxb commented 2 years ago

Hi @LiyaoTang

How to set the GPU device to revise the CUDA_VISIBLE_DEVICES? not only 0 1, can also use 0 2. Thanks.

LiyaoTang commented 2 years ago

you can set the device with comma separated string, i.e. --gpu 0,2 in your situation.