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.

Sign up for GitHub

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

Jump to bottom

Where is the CUDA path? #6

Closed
QiangZhangCV opened this issue Feb 12, 2023 · 2 comments
Closed

Where is the CUDA path? #6

QiangZhangCV opened this issue Feb 12, 2023 · 2 comments

Comments

@QiangZhangCV
Copy link

Thanks for your great work!
I am using the following docker image:
cnstark/pytorch:1.12.0-py3.9.12-cuda11.6.2-ubuntu20.04

I can not found the cuda path. /usr/local/cuda not exisits. As onnxruntime, TensorRT need the cuda well installed, could you provide more informations about this?
@cnstark
Copy link
Owner

cnstark commented Feb 15, 2023

Hi!
This version does not contain CUDA files, it is built on the standard Ubuntu image.
Please use cnstark/pytorch:1.12.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04 instead of the current image.

@cnstark cnstark closed this as completed Feb 16, 2023
@QiangZhangCV
Copy link
Author

Right, Got it. Thanks very much!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@QiangZhangCV @cnstark