Error “No module named 'torchvision'” when running .dlpk model in ArcGIS Pro 3.5.3

[mari_rodrigues]

Hi everyone! 😊

I'm trying to run a Deep Learning model (.dlpk) using the Detect Objects Using Deep Learning tool in ArcGIS Pro 3.5.3, and I'm getting the following error:

error image 

Error 002667: Unable to initialize python raster function with scalar arguments.
ModuleNotFoundError: No module named 'torchvision'

The .dlpk model was exported correctly and works on other machines,
so I believe the issue might be related to the local Python environment or missing libraries.

Could you please help me understand:

1. What is the correct way to configure the Python environment for running .dlpk models in ArcGIS Pro 3.5.3?

2. Does the torchvision package come included with Esri’s Deep Learning Framework, or should it be installed manually?

3. Is there any recent documentation or recommendation for this specific error?

Thank you very much in advance!
Mariana

[KeiranBray]

Hey @mari_rodrigues did you download and install the deep learning frameworks for ArcGIS Pro on the machine you're using?

Torchvision is included in the deep learning frameworks and is listed here: Esri/deep-learning-frameworks: Installation support for Deep Learning Frameworks for the ArcGIS Syst... this resource also has the install files.

Another spot to check if you did install the frameworks is maybe the package manager in ArcGIS Pro, and check to see which is the active environment.