'torchvision.version' has no attribute named 'cuda'

jmwolf27 · September 21, 2023, 4:42pm

Hi,
I have followed the Pytorch and torchvision installation steps from both the ORIN setup guide and various posts on the developer forums. The problem that I am having is getting torchvision to install with cuda on the ORIN. When I used the L4T containers provided, the torchvision installation there has the cuda attribute. I am confused as to why there is a discrepency and would like to know how to get the cuda attribute to be installed and working with torchvision. Please help.
Regards,
Jordan

dusty_nv · September 21, 2023, 10:47pm

Hi Jordan, here’s the procedure that’s done in the dockerfile to install torchvision on top of PyTorch:

github.com

dusty-nv/jetson-containers/blob/master/packages/pytorch/torchvision/Dockerfile

#---
# name: torchvision
# group: ml
# config: config.py
# depends: [cmake, pytorch]
# test: test.py
#---
ARG BASE_IMAGE
FROM ${BASE_IMAGE}

ARG TORCHVISION_VERSION
ARG TORCH_CUDA_ARCH_LIST="5.3;6.2;7.2;8.7"

RUN printenv && echo "torchvision version = $TORCHVISION_VERSION" && echo "TORCH_CUDA_ARCH_LIST = $TORCH_CUDA_ARCH_LIST"

RUN apt-get update && \
    apt-get install -y --no-install-recommends \
            libjpeg-dev \
		  zlib1g-dev \
    && rm -rf /var/lib/apt/lists/* \

This file has been truncated. show original

You should be able to follow those same steps inside the RUN commands.

jmwolf27 · September 22, 2023, 2:07pm

Thanks @dusty_nv! That did the trick. I did have to remove torchvision from the base environment (think this might have caused some issues) and then rebuild it in the virtual environment. When I did such, I was able to verify that torchvision indeed is the cuda version.

jmwolf27 · October 3, 2023, 4:33pm

@dusty_nv some additional details:

Running the script in a new environment didn’t work. I dug into torchvision’s setup.py file to find a solution. I found that by including the following in the script:

export FORCE_CUDA=1
export CUDA_HOME=/usr/local/cuda-11/

We would essentially force the cuda installation of torchvision. From here the torchvision script became reproducible in newly created environments.

system · October 17, 2023, 4:34pm

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.