Closed sxhxliang closed 4 years ago
Your GPU needs to be available inside Docker
Try docker run --gpus all nvidia/cuda:10.0-base nvidia-smi
to verify.
If that fails follow the installation guidelines here
For triton you need at least nvidia driver 450 (440 if you have a tesla based GPU like the T4). Like you can see in your nvidia-smi
you have 430.34 which is not high enough!
https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html
@sxhxliang Could you resolve the issue by upgrading your driver version and installing nvidia-docker?
@sxhxliang Could you resolve the issue by upgrading your driver version and installing nvidia-docker?
yes
I get a CUDA error while executing the
main
my ubuntu