Linux

CUDA

Compatible with NVIDIA graphic cards:

conda install conda-forge::cuda-runtime=12.8.0 conda-forge::cudnn=9.7.1.26

TensorRT

Suitable for high performance NVIDIA graphic cards:

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

ROCm

Compatible with AMD graphic cards:

curl --output amdgpu-install-6.3.1.deb https://repo.radeon.com/amdgpu-install/6.3.1/ubuntu/jammy/amdgpu-install_6.3.60301-1_all.deb

dpkg --install amdgpu-install-6.3.1.deb

conda install conda-forge::rocm-smi=6.1.2 conda-forge::gcc=14.2.0

OpenVINO

Compatible with Intel graphic cards:

conda install conda-forge::openvino=2025.0.0