Ray

How to train by a single gpu cuda:1 怎样只用cuda：1单gpu训练

2

**I have changed the code to "self.net = MMDataParallel(self.net,device_ids=[self.cfg.gpus]).cuda(self.cfg.gpus)", but there is a mistake** Validate: 0%| | 0/70 [05:49

I'm having some problems of Error :python setup.py build develop

5

"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin\nvcc" -c ./clrnet/ops/csrc\nms_kernel.cu -o build\temp.win-amd64-3.9\Release\./clrnet/ops/csrc\nms_kernel.obj -IH:\ana\envs\swinunet\lib\site-packages\torch\include -IH:\ana\envs\swinunet\lib\site-packages\torch\include\torch\csrc\api\include -IH:\ana\envs\swinunet\lib\site-packages\torch\include\TH -IH:\ana\envs\swinunet\lib\site-packages\torch\include\THC "-IC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include" -IH:\ana\envs\swinunet\include -IH:\ana\envs\swinunet\Include "-IH:\Microsoft\Microsoft Visual Studio\2022\Enterprise\VC\Tools\MSVC\14.32.31326\ATLMFC\include" "-IH:\Microsoft\Microsoft Visual Studio\2022\Enterprise\VC\Tools\MSVC\14.32.31326\include" "-IH:\Windows Kits\10\include\10.0.19041.0\ucrt" "-IH:\Windows Kits\10\\include\10.0.19041.0\\shared"...