sci-ml/caffe2-rocm
A deep learning framework (ROCm backend)
Screenshots
-
caffe2-rocm-2.12.0~amd64cusparselt distributed fbgemm flash gloo kineto memefficient mimalloc mkl mpi nccl nnpack +numpy onednn openblas opencl openmp qnnpack xnnpack python_single_target_python3_12 python_single_target_python3_13 python_single_target_python3_14 +amdgpu_targets_gfx908 +amdgpu_targets_gfx90a +amdgpu_targets_gfx942 +amdgpu_targets_gfx1030 +amdgpu_targets_gfx1100 +amdgpu_targets_gfx1101 +amdgpu_targets_gfx1200 +amdgpu_targets_gfx1201 amdgpu_targets_gfx803 amdgpu_targets_gfx900 amdgpu_targets_gfx906 amdgpu_targets_gfx940 amdgpu_targets_gfx941 amdgpu_targets_gfx1010 amdgpu_targets_gfx1011 amdgpu_targets_gfx1012 amdgpu_targets_gfx1031 amdgpu_targets_gfx1102 amdgpu_targets_gfx1103 amdgpu_targets_gfx1150 amdgpu_targets_gfx1151
View 
Download 
Browse License: BSD Overlay: zGentoo
Runtime Dependencies
caffe2-rocm-2.12.0
python_single_target_python3_12?
( dev-lang/python:3.12 )
python_single_target_python3_13?
( dev-lang/python:3.13 )
python_single_target_python3_14?
( dev-lang/python:3.14 )
app-eselect/eselect-caffe2
dev-cpp/abseil-cpp:=
dev-cpp/gflags:=
>=dev-cpp/glog-0.5.0:=
>=dev-libs/cpuinfo-2025.11.14
dev-libs/libfmt:=
dev-libs/protobuf:=
dev-libs/sleef
sci-ml/onnx
virtual/lapack
nccl?
( >=dev-libs/rccl-6.3:= <dev-libs/rccl-7.3:= )
>=dev-util/hip-6.3:=
<dev-util/hip-7.3:=
>=dev-util/roctracer-6.3:=
<dev-util/roctracer-7.3:=
>=sci-libs/hipBLAS-6.3:=
<sci-libs/hipBLAS-7.3:=[rocsolver
(+)
]
>=sci-libs/hipBLASLt-6.3:=
<sci-libs/hipBLASLt-7.3:=
>=sci-libs/hipFFT-6.3:=
<sci-libs/hipFFT-7.3:=
>=sci-libs/hipRAND-6.3:=
<sci-libs/hipRAND-7.3:=
>=sci-libs/hipSOLVER-6.3:=
<sci-libs/hipSOLVER-7.3:=
>=sci-libs/hipSPARSE-6.3:=
<sci-libs/hipSPARSE-7.3:=
>=sci-libs/miopen-6.3:=
<sci-libs/miopen-7.3:=
>=sci-libs/rocBLAS-6.3:=
<sci-libs/rocBLAS-7.3:=
>=sci-libs/rocRAND-6.3:=
<sci-libs/rocRAND-7.3:=
>=sci-libs/rocSOLVER-6.3:=
<sci-libs/rocSOLVER-7.3:=
memefficient?
( =sci-libs/aotriton-bin-0.11*:= )
distributed?
( >=dev-util/rocm-smi-6.3:= <dev-util/rocm-smi-7.3:= >=dev-util/amdsmi-6.3:= <dev-util/amdsmi-7.3:= dev-cpp/cpp-httplib:= )
cusparselt?
( >=sci-libs/hipsparselt-6.3:= <sci-libs/hipsparselt-7.3:= )
fbgemm?
( >=sci-ml/FBGEMM-1.4 )
gloo?
( >=sci-ml/gloo-2025.06.04[rocm] )
kineto?
( ~sci-ml/kineto-0.4.0_p20260323 )
mimalloc?
( dev-libs/mimalloc )
mpi?
( virtual/mpi )
nnpack?
( sci-ml/NNPACK dev-libs/pthreadpool )
numpy?
( python_single_target_python3_12?
( dev-python/numpy[python_targets_python3_12
(-)
] )
python_single_target_python3_13?
( dev-python/numpy[python_targets_python3_13
(-)
] )
python_single_target_python3_14?
( dev-python/numpy[python_targets_python3_14
(-)
] )
)
onednn?
( sci-ml/oneDNN )
opencl?
( virtual/opencl )
qnnpack?
( !sci-libs/QNNPACK sci-ml/gemmlowp dev-libs/pthreadpool )
xnnpack?
( >=sci-ml/XNNPACK-2024.11 dev-libs/pthreadpool )
mkl?
( sci-libs/mkl )
openblas?
( sci-libs/openblas )