CUDA not found on DJI Manifold 2G
453 0 2022-1-2
Uploading and Loding Picture ...(0/1)
o(^-^)o
pylonicGateway
lvl.1
Singapore
Offline

Hello everyone. I am trying to run a github repo with torch on the manifold 2G.
The repo is from: https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
I have gotten this error




Loading weights from deep_sort_pytorch/deep_sort/deep/checkpoint/ckpt.t7... Done!
Traceback (most recent call last):
  File "track.py", line 243, in <module>
    detect(opt)
  File "track.py", line 51, in detect
    device = select_device(opt.device)
  File "/media/dji/80GBstore/targetTrack/mikelbrostromNov2021/Yolov5_DeepSort_Pytorch/yolov5/utils/torch_utils.py", line 65, in select_device
    assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested'  # check availability
AssertionError: CUDA unavailable, invalid device 0 requested



I understand that with Manifold 2G, there is jetpack 3.3 installed, which contains CUDA 9 !However i cannot install CUDA enabled pytorch version as well.Here are some of the attempts


pip install torch-1.1.0-cp37-cp37m-linux_x86_64.whl
ERROR: torch-1.1.0-cp37-cp37m-linux_x86_64.whl is not a supported wheel on this platform.

pip install torch-1.0.1.post2-cp37-cp37m-linux_x86_64.whl
ERROR: torch-1.0.1.post2-cp37-cp37m-linux_x86_64.whl is not a supported wheel on this platform.

pip install torch-1.7.0+cu92-cp38-cp38-linux_x86_64.whl
ERROR: torch-1.7.0+cu92-cp38-cp38-linux_x86_64.whl is not a supported wheel on this platform.

pip install torch-1.7.1+cu92-cp39-cp39-linux_x86_64.whl
ERROR: torch-1.7.1+cu92-cp39-cp39-linux_x86_64.whl is not a supported wheel on this platform.

pip install torch-1.7.1+cpu-cp39-cp39-linux_x86_64.whl
ERROR: torch-1.7.1+cpu-cp39-cp39-linux_x86_64.whl is not a supported wheel on this platform.



I am trying to install the torch version on python virtual environments (non-Conda), and here are the terminal outptus when i call python3.7 and python3.8

Python 3.8.9 (default, Apr  3 2021, 01:02:10)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.


Python 3.7.10 (default, Feb 20 2021, 21:21:24)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.



Could anyone advise what could be the problem ?
1) Why cant CUDA-compatible version of torch be installed ?

2) Are there any updates I have to do before this can be done ?




2022-1-2
Use props
Advanced
You need to log in before you can reply Login | Register now

Credit Rules