I’m trying to profile code using the bottleneck util. Running CUDA_VISIBLE_DEVICES=2 python3 -m torch.utils.bottleneck /root/mount/Matterport3DSimulator/scripts/multimodal_filter.py --mode eval --batch_size 3000 --checkpoint best_model.pt, gives the following error:

image1898×445 28.9 KB

The class its referring to is a class declared in the multimodal_filter script. That script works with no issues.
I’m using docker (nvidia-docker), PyTorch 1.4.0, Cuda compilation tools, release 10.0 (I’m assuming this is the cuda version).