Linux

CUDA

Compatible with NVIDIA graphic cards:

conda install nvidia/label/cuda-12.9.1::cuda-runtime nvidia/label/cudnn-9.10.0::cudnn

TensorRT

Suitable for high performance NVIDIA graphic cards:

pip install tensorrt==10.12.0.36 --extra-index-url https://pypi.nvidia.com

ROCm

Compatible with AMD graphic cards:

curl --output amdgpu-install-7.2.deb https://repo.radeon.com/amdgpu-install/7.2/ubuntu/jammy/amdgpu-install_7.2.70200-1_all.deb
dpkg --install amdgpu-install-7.2.deb
conda install conda-forge::gcc=15.2.0

OpenVINO

Compatible with Intel graphic cards:

conda install conda-forge::openvino=2025.3.0

Last updated