Traceback (most recent call last):


 "test_torch_encoding.py", line 15, in <module>

output = model.evaluate(img)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch_encoding-1.2.2b20210130-py3.8-linux-x86_64.egg/encoding/models/sseg/base.py", line 101, in evaluate

pred = self.forward(x)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch_encoding-1.2.2b20210130-py3.8-linux-x86_64.egg/encoding/models/sseg/deeplab.py", line 47, in forward

c1, c2, c3, c4 = self.base_forward(x)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch_encoding-1.2.2b20210130-py3.8-linux-x86_64.egg/encoding/models/sseg/base.py", line 96, in base_forward

c3 = self.pretrained.layer3(c2)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/container.py", line 117, in forward

input = module(input)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch_encoding-1.2.2b20210130-py3.8-linux-x86_64.egg/encoding/models/backbone/resnet.py", line 95, in forward

out = self.conv2(out)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch_encoding-1.2.2b20210130-py3.8-linux-x86_64.egg/encoding/nn/splat.py", line 50, in forward

x = self.bn0(x)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/modules/batchnorm.py", line 131, in forward

return F.batch_norm(


 "/home/mona/venv/torchenc/lib/python3.8/site-packages/torch/nn/functional.py", line 2056, in batch_norm

return torch.batch_norm(
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.82 GiB total capacity; 2.20 GiB already allocated; 27.88 MiB free; 2.32 GiB reserved in total by PyTorch)


Traceback (most recent call last):


 "/usr/local/lib/python3.7/site-packages/pip/_vendor/pep517/_in_process.py", line 280, in <module>

main()


 "/usr/local/lib/python3.7/site-packages/pip/_vendor/pep517/_in_process.py", line 263, in main

json_out['return_val'] = hook(**hook_input['kwargs'])


 "/usr/local/lib/python3.7/site-packages/pip/_vendor/pep517/_in_process.py", line 205, in build_wheel

metadata_directory)


 "/tmp/pip-build-env-r_fmi6tv/overlay/lib/python3.7/site-packages/setuptools/build_meta.py", line 209, in build_wheel

wheel_directory, config_settings)


 "/tmp/pip-build-env-r_fmi6tv/overlay/lib/python3.7/site-packages/setuptools/build_meta.py", line 194, in _build_with_temp_dir

self.run_setup()


 "/tmp/pip-build-env-r_fmi6tv/overlay/lib/python3.7/site-packages/setuptools/build_meta.py", line 142, in run_setup

exec(compile(code, __file__, 'exec'), locals())


 "setup.py", line 101, in <module>

zip_safe=False,


 "/tmp/pip-build-env-r_fmi6tv/overlay/lib/python3.7/site-packages/setuptools/__init__.py", line 145, in setup

return distutils.core.setup(**attrs)


 "/usr/local/lib/python3.7/distutils/core.py", line 148, in setup

dist.run_commands()


 "/usr/local/lib/python3.7/distutils/dist.py", line 966, in run_commands

self.run_command(cmd)


 "/usr/local/lib/python3.7/distutils/dist.py", line 985, in run_command

cmd_obj.run()


 "/tmp/pip-build-env-r_fmi6tv/overlay/lib/python3.7/site-packages/wheel/bdist_wheel.py", line 228, in run

self.run_command('install')


 "/usr/local/lib/python3.7/distutils/cmd.py", line 313, in run_command

self.distribution.run_command(command)


 "/usr/local/lib/python3.7/distutils/dist.py", line 985, in run_command

cmd_obj.run()


 "setup.py", line 55, in run

"cargo not found in PATH. Please install rust "

RuntimeError: cargo not found in PATH. Please install rust (https://www.rust-lang.org/tools/install) and try again
