Closed VickyCas closed 1 year ago
--gpu-id
to -1
to deploy models on the CPUs.
- If you find only the agents in the first launched screen do not move, it is OK.
- Can you report the version of your PyTorch and GPU? I guess your crash may be caused by the out-of-memory of the GPU. You can set the
--gpu-id
to-1
to deploy models on the CPUs.
Thank you for your reply! The version is below:
print(torch.version) 1.2.0
nvidia-smi -L GPU 0: NVIDIA GeForce RTX 3090 (UUID: GPU-1b5eb693-4789-060a-f54f-84d6f11d2b44)
Please try the newer version of PyTorch (e.g., 1.7 with CUDA 11), as RTX 3090 does not support the old version of CUDA.
Hi @zfw1226 ,when i run your code https://github.com/zfw1226/active_tracking_rl/tree/distractor, i met some errors: