Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update requirements_gpu.txt #22

Merged
merged 1 commit into from
Dec 10, 2021
Merged

Conversation

sam-writer
Copy link
Contributor

The last update added from pytorch_quantization.nn import TensorQuantizer but didn't add the dependency

@pommedeterresautee
Copy link
Member

On my machine I have to install it from source, does pip install work out of the box on yours?

@sam-writer
Copy link
Contributor Author

Yeah, on the VM I am using it worked to just pip install it.

Ubuntu 20.04.3

nvidia-cublas-cu11==2021.10.25
nvidia-cublas-cu115==11.7.3.1
nvidia-cuda-runtime-cu11==2021.10.25
nvidia-cuda-runtime-cu115==11.5.50
nvidia-cudnn-cu11==2021.11.18
nvidia-cudnn-cu115==8.3.0.98
nvidia-pyindex==1.0.9
nvidia-tensorrt==8.2.1.8
onnx==1.10.2
onnx-graphsurgeon==0.3.14
onnxruntime-gpu==1.9.0

pytorch-quantization==2.1.2

torch==1.10.0+cu111

Copy link
Member

@pommedeterresautee pommedeterresautee left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm
@sam-writer you were right it's an issue on my machine.
Tks!

@pommedeterresautee pommedeterresautee merged commit 75f1d52 into ELS-RD:main Dec 10, 2021
@pommedeterresautee pommedeterresautee added bug Something isn't working TensorRT labels Dec 10, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working TensorRT
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants