sai@sai-desktop:~$ wget https://developer.download.nvidia.cn/compute/redist/jp/v60dp/pytorch/torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl -O torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
--2024-01-29 11:11:16-- https://developer.download.nvidia.cn/compute/redist/jp/v60dp/pytorch/torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl
Resolving developer.download.nvidia.cn (developer.download.nvidia.cn)... 107.151.190.158, 107.151.190.157
Connecting to developer.download.nvidia.cn (developer.download.nvidia.cn)|107.151.190.158|:443... connected.
HTTP request sent, awaiting response... 200 OK
Length: 921771811 (879M) [application/octet-stream]
Saving to: ‘torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl’
torch-2.2.0a0+6a974be.nv23.11-c 100%[=====================================================>] 879.07M 42.0MB/s in 26s
2024-01-29 11:11:43 (33.3 MB/s) - ‘torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl’ saved [921771811/921771811]
Reading package lists... Done
Building dependency tree... Done
Reading state information... Done
libomp-dev is already the newest version (1:14.0-55~exp2).
libopenblas-base is already the newest version (0.3.20+ds-1).
libopenmpi-dev is already the newest version (4.1.2-2ubuntu1).
python3-pip is already the newest version (22.0.2+dfsg-1ubuntu0.4).
0 upgraded, 0 newly installed, 0 to remove and 0 not upgraded.
sai@sai-desktop:~$ sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
Reading package lists... Done
Building dependency tree... Done
Reading state information... Done
libomp-dev is already the newest version (1:14.0-55~exp2).
libopenblas-base is already the newest version (0.3.20+ds-1).
libopenmpi-dev is already the newest version (4.1.2-2ubuntu1).
python3-pip is already the newest version (22.0.2+dfsg-1ubuntu0.4).
0 upgraded, 0 newly installed, 0 to remove and 0 not upgraded.
sai@sai-desktop:~$ pip3 install Cython==0.29.36
pip3 install numpy torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl
Defaulting to user installation because normal site-packages is not writeable
Requirement already satisfied: Cython==0.29.36 in ./.local/lib/python3.10/site-packages (0.29.36)
Defaulting to user installation because normal site-packages is not writeable
Processing ./torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl
Requirement already satisfied: numpy in /usr/lib/python3/dist-packages (1.21.5)
Requirement already satisfied: sympy in /usr/lib/python3/dist-packages (from torch==2.2.0a0+6a974be.nv23.11) (1.9)
Requirement already satisfied: filelock in ./.local/lib/python3.10/site-packages (from torch==2.2.0a0+6a974be.nv23.11) (3.13.1)
Requirement already satisfied: jinja2 in ./.local/lib/python3.10/site-packages (from torch==2.2.0a0+6a974be.nv23.11) (3.1.3)
Requirement already satisfied: networkx in ./.local/lib/python3.10/site-packages (from torch==2.2.0a0+6a974be.nv23.11) (3.2.1)
Requirement already satisfied: typing-extensions in ./.local/lib/python3.10/site-packages (from torch==2.2.0a0+6a974be.nv23.11) (4.9.0)
Requirement already satisfied: fsspec in ./.local/lib/python3.10/site-packages (from torch==2.2.0a0+6a974be.nv23.11) (2023.12.2)
Requirement already satisfied: MarkupSafe>=2.0 in /usr/lib/python3/dist-packages (from jinja2->torch==2.2.0a0+6a974be.nv23.11) (2.0.1)
Installing collected packages: torch
Successfully installed torch-2.2.0a0+6a974be
sai@sai-desktop:~$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.16 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
Reading package lists... Done
Building dependency tree... Done
Reading state information... Done
libjpeg-dev is already the newest version (8c-2ubuntu10).
libopenblas-dev is already the newest version (0.3.20+ds-1).
libpython3-dev is already the newest version (3.10.6-1~22.04).
zlib1g-dev is already the newest version (1:1.2.11.dfsg-2ubuntu9.2).
libavcodec-dev is already the newest version (7:4.4.2-0ubuntu0.22.04.1).
libavformat-dev is already the newest version (7:4.4.2-0ubuntu0.22.04.1).
libswscale-dev is already the newest version (7:4.4.2-0ubuntu0.22.04.1).
0 upgraded, 0 newly installed, 0 to remove and 0 not upgraded.
fatal: destination path 'torchvision' already exists and is not an empty directory.
sai@sai-desktop:~$ git clone --branch v0.16 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
Cloning into 'torchvision'...
fatal: Remote branch v0.16 not found in upstream origin
sai@sai-desktop:~$ git clone --branch v0.16.0 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
Cloning into 'torchvision'...
remote: Enumerating objects: 461195, done.
remote: Counting objects: 100% (41654/41654), done.
remote: Compressing objects: 100% (1956/1956), done.
remote: Total 461195 (delta 39726), reused 41505 (delta 39636), pack-reused 419541
Receiving objects: 100% (461195/461195), 905.67 MiB | 22.23 MiB/s, done.
Resolving deltas: 100% (428829/428829), done.
Note: switching to 'fbb4cc54ed521ba912f50f180dc16a213775bf5c'.
You are in 'detached HEAD' state. You can look around, make experimental
changes and commit them, and you can discard any commits you make in this
state without impacting any branches by switching back to a branch.
If you want to create a new branch to retain commits you create, you may
do so (now or later) by using -c with the switch command. Example:
git switch -c <new-branch-name>
Or undo this operation with:
git switch -
Turn off this advice by setting config variable advice.detachedHead to false
sai@sai-desktop:~$ cd torchvision
export BUILD_VERSION=0.16.0 # where 0.x.0 is the torchvision version
sai@sai-desktop:~/torchvision$ python3 setup.py install --user
/home/sai/torchvision/setup.py:1: DeprecationWarning: The distutils package is deprecated and slated for removal in Python 3.12. Use setuptools or check PEP 632 for potential alternatives
import distutils.command.clean
/home/sai/.local/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: module compiled against API version 0x10 but this version of numpy is 0xe . Check the section C-API incompatibility at the Troubleshooting ImportError section at https://numpy.org/devdocs/user/troubleshooting-importerror.html#c-api-incompatibility for indications on how to solve this problem . (Triggered internally at /opt/pytorch/pytorch/torch/csrc/utils/tensor_numpy.cpp:84.)
device: torch.device = torch.device(torch._C._get_default_device()), # torch.device('cpu'),
Building wheel torchvision-0.16.0
Compiling extensions with following flags:
FORCE_CUDA: False
FORCE_MPS: False
DEBUG: False
TORCHVISION_USE_PNG: True
TORCHVISION_USE_JPEG: True
TORCHVISION_USE_NVJPEG: True
TORCHVISION_USE_FFMPEG: True
TORCHVISION_USE_VIDEO_CODEC: True
NVCC_FLAGS:
Compiling with debug mode OFF
Building torchvision without PNG image support
Running build on conda-build: False
Running build on conda: False
Building torchvision with JPEG image support
libjpeg include path: None
libjpeg lib path: None
Building torchvision with NVJPEG image support
Building torchvision without ffmpeg support
Building torchvision without video codec support
running install
/usr/lib/python3/dist-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools.
warnings.warn(
/usr/lib/python3/dist-packages/setuptools/command/easy_install.py:158: EasyInstallDeprecationWarning: easy_install command is deprecated. Use build and pip and other standards-based tools.
warnings.warn(
running bdist_egg
running egg_info
creating torchvision.egg-info
writing torchvision.egg-info/PKG-INFO
writing dependency_links to torchvision.egg-info/dependency_links.txt
writing requirements to torchvision.egg-info/requires.txt
writing top-level names to torchvision.egg-info/top_level.txt
writing manifest file 'torchvision.egg-info/SOURCES.txt'
/home/sai/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py:502: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.
warnings.warn(msg.format('we could not find ninja.'))
reading manifest file 'torchvision.egg-info/SOURCES.txt'
reading manifest template 'MANIFEST.in'
warning: no previously-included files matching '__pycache__' found under directory '*'
warning: no previously-included files matching '*.py[co]' found under directory '*'
adding license file 'LICENSE'
writing manifest file 'torchvision.egg-info/SOURCES.txt'
installing library code to build/bdist.linux-aarch64/egg
running install_lib
running build_py
creating build
creating build/lib.linux-aarch64-3.10
creating build/lib.linux-aarch64-3.10/torchvision
copying torchvision/_meta_registrations.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/extension.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/_internally_replaced_utils.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/__init__.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/_utils.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/version.py -> build/lib.linux-aarch64-3.10/torchvision
copying torchvision/utils.py -> build/lib.linux-aarch64-3.10/torchvision
creating build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/dtd.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/_optical_flow.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/ucf101.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/inaturalist.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/clevr.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/sbu.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/flowers102.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/places365.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/lfw.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/food101.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/widerface.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/video_utils.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/kitti.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/fgvc_aircraft.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/eurosat.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/coco.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/sun397.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/phototour.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/sbd.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/rendered_sst2.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/vision.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/lsun.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/cifar.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/stl10.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/voc.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/usps.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/caltech.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/flickr.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/celeba.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/fer2013.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/oxford_iiit_pet.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/mnist.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/folder.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/pcam.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/fakedata.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/hmdb51.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/svhn.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/omniglot.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/cityscapes.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/_stereo_matching.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/kinetics.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/moving_mnist.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/country211.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/stanford_cars.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/gtsrb.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/semeion.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/utils.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
copying torchvision/datasets/imagenet.py -> build/lib.linux-aarch64-3.10/torchvision/datasets
creating build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/vgg.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/inception.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/resnet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/swin_transformer.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/vision_transformer.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/_api.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/regnet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/mobilenet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/googlenet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/densenet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/mobilenetv3.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/squeezenet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/convnext.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/alexnet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/mnasnet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/feature_extraction.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/mobilenetv2.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/_meta.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/efficientnet.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/shufflenetv2.py -> build/lib.linux-aarch64-3.10/torchvision/models
copying torchvision/models/maxvit.py -> build/lib.linux-aarch64-3.10/torchvision/models
creating build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/boxes.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/ps_roi_pool.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/_register_onnx_ops.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/diou_loss.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/focal_loss.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/misc.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/drop_block.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/poolers.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/giou_loss.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/roi_align.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/deform_conv.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/_box_convert.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/ciou_loss.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/roi_pool.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/stochastic_depth.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/ps_roi_align.py -> build/lib.linux-aarch64-3.10/torchvision/ops
copying torchvision/ops/feature_pyramid_network.py -> build/lib.linux-aarch64-3.10/torchvision/ops
creating build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/functional.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/transforms.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/_functional_tensor.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/autoaugment.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/functional_tensor.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/_presets.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/functional_pil.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/_functional_pil.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/_transforms_video.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
copying torchvision/transforms/_functional_video.py -> build/lib.linux-aarch64-3.10/torchvision/transforms
creating build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_bounding_boxes.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_image.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_torch_function_helpers.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_dataset_wrapper.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_mask.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_tv_tensor.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
copying torchvision/tv_tensors/_video.py -> build/lib.linux-aarch64-3.10/torchvision/tv_tensors
creating build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/video.py -> build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/_load_gpu_decoder.py -> build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/video_reader.py -> build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/_video_opt.py -> build/lib.linux-aarch64-3.10/torchvision/io
copying torchvision/io/image.py -> build/lib.linux-aarch64-3.10/torchvision/io
creating build/lib.linux-aarch64-3.10/torchvision/datasets/samplers
copying torchvision/datasets/samplers/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/datasets/samplers
copying torchvision/datasets/samplers/clip_sampler.py -> build/lib.linux-aarch64-3.10/torchvision/datasets/samplers
creating build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/keypoint_rcnn.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/retinanet.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/faster_rcnn.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/backbone_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/roi_heads.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/generalized_rcnn.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/ssd.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/ssdlite.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/fcos.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/anchor_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/image_list.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/mask_rcnn.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/rpn.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
copying torchvision/models/detection/transform.py -> build/lib.linux-aarch64-3.10/torchvision/models/detection
creating build/lib.linux-aarch64-3.10/torchvision/models/video
copying torchvision/models/video/resnet.py -> build/lib.linux-aarch64-3.10/torchvision/models/video
copying torchvision/models/video/swin_transformer.py -> build/lib.linux-aarch64-3.10/torchvision/models/video
copying torchvision/models/video/s3d.py -> build/lib.linux-aarch64-3.10/torchvision/models/video
copying torchvision/models/video/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models/video
copying torchvision/models/video/mvit.py -> build/lib.linux-aarch64-3.10/torchvision/models/video
creating build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/inception.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/resnet.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/mobilenet.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/googlenet.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/mobilenetv3.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/mobilenetv2.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/shufflenetv2.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
copying torchvision/models/quantization/utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/quantization
creating build/lib.linux-aarch64-3.10/torchvision/models/segmentation
copying torchvision/models/segmentation/deeplabv3.py -> build/lib.linux-aarch64-3.10/torchvision/models/segmentation
copying torchvision/models/segmentation/fcn.py -> build/lib.linux-aarch64-3.10/torchvision/models/segmentation
copying torchvision/models/segmentation/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models/segmentation
copying torchvision/models/segmentation/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/segmentation
copying torchvision/models/segmentation/lraspp.py -> build/lib.linux-aarch64-3.10/torchvision/models/segmentation
creating build/lib.linux-aarch64-3.10/torchvision/models/optical_flow
copying torchvision/models/optical_flow/raft.py -> build/lib.linux-aarch64-3.10/torchvision/models/optical_flow
copying torchvision/models/optical_flow/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/models/optical_flow
copying torchvision/models/optical_flow/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/models/optical_flow
creating build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_container.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_type_conversion.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_transform.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_color.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_auto_augment.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_deprecated.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_meta.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_augment.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_misc.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_geometry.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
copying torchvision/transforms/v2/_temporal.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2
creating build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_type_conversion.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_color.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/__init__.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_utils.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_deprecated.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_meta.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_augment.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_misc.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_geometry.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
copying torchvision/transforms/v2/functional/_temporal.py -> build/lib.linux-aarch64-3.10/torchvision/transforms/v2/functional
running build_ext
Traceback (most recent call last):
File "/home/sai/torchvision/setup.py", line 542, in <module>
setup(
File "/usr/lib/python3/dist-packages/setuptools/__init__.py", line 153, in setup
return distutils.core.setup(**attrs)
File "/usr/lib/python3.10/distutils/core.py", line 148, in setup
dist.run_commands()
File "/usr/lib/python3.10/distutils/dist.py", line 966, in run_commands
self.run_command(cmd)
File "/usr/lib/python3.10/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/install.py", line 74, in run
self.do_egg_install()
File "/usr/lib/python3/dist-packages/setuptools/command/install.py", line 116, in do_egg_install
self.run_command('bdist_egg')
File "/usr/lib/python3.10/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/usr/lib/python3.10/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/bdist_egg.py", line 164, in run
cmd = self.call_command('install_lib', warn_dir=0)
File "/usr/lib/python3/dist-packages/setuptools/command/bdist_egg.py", line 150, in call_command
self.run_command(cmdname)
File "/usr/lib/python3.10/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/usr/lib/python3.10/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/install_lib.py", line 23, in run
self.build()
File "/usr/lib/python3.10/distutils/command/install_lib.py", line 109, in build
self.run_command('build_ext')
File "/usr/lib/python3.10/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/usr/lib/python3.10/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/build_ext.py", line 79, in run
_build_ext.run(self)
File "/home/sai/.local/lib/python3.10/site-packages/Cython/Distutils/old_build_ext.py", line 186, in run
_build_ext.build_ext.run(self)
File "/usr/lib/python3.10/distutils/command/build_ext.py", line 340, in run
self.build_extensions()
File "/home/sai/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 525, in build_extensions
_check_cuda_version(compiler_name, compiler_version)
File "/home/sai/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 413, in _check_cuda_version
raise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda))
RuntimeError:
The detected CUDA version (11.5) mismatches the version that was used to compile
PyTorch (12.2). Please make sure to use the same CUDA versions.
Hi @sairaghava2013,
You can try to manually compile it, you can check here for more details.
Regards,
Andres
Embedded SW Engineer at RidgeRun
Contact us: support@ridgerun.com
Developers wiki: https://developer.ridgerun.com/
Website: www.ridgerun.com
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.