You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hello,
An error occurred while installing torch2trt
My environment is
Ubuntu 18.04
CUDA 11.7
CUDNN 8.7.0
Python 3.9.16
Torch 1.13.1 (torchvision==0.14.1)
and using NVIDIA RTX 3060 as GPU.
As far as I know torch2trt uses ROCm, an AMD GPU-specific driver for conversion. However, as you can see, I am not using an AMD GPU.
Is there any way to fix this error?
Additionally, I was using an earlier version of CUDA 11.0, Python 3.6, and Torch (maybe <= 1.9), and I was able to use torch2trt without any issues. However, I needed to update Torch, so I updated my environment above. Could this have caused the issue? What is the possibility of a collision occurring?
The text was updated successfully, but these errors were encountered:
Hello,
An error occurred while installing torch2trt
My environment is
As far as I know torch2trt uses ROCm, an AMD GPU-specific driver for conversion. However, as you can see, I am not using an AMD GPU.
Is there any way to fix this error?
Additionally, I was using an earlier version of CUDA 11.0, Python 3.6, and Torch (maybe <= 1.9), and I was able to use torch2trt without any issues. However, I needed to update Torch, so I updated my environment above. Could this have caused the issue? What is the possibility of a collision occurring?
The text was updated successfully, but these errors were encountered: