ROCm / MIVisionX

MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into a single toolkit. AMD MIVisionX also delivers a highly optimized open-source implementation of the Khronos OpenVX™ and OpenVX™ Extensions.

Home Page:https://rocm.docs.amd.com/projects/MIVisionX/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Setup script - Add CuPy

LakshmiKumar23 opened this issue · comments

Add CuPy to all dockerfiles and setup script for rocAL
Instructions:

apt-get update
apt install -y hipblas hipsparse rocrand rocthrust hipcub git g++ hipfft rocfft python3-dev

#Install CuPy
git clone https://github.com/ROCmSoftwarePlatform/cupy.git && cd cupy
export CUPY_INSTALL_USE_HIP=1
export HCC_AMDGPU_TARGET=gfx90a (or need to identify and add any other target architectures comma-separated)
export ROCM_HOME=/opt/rocm
git submodule update --init
pip install -e . --no-cache-dir -vvvv

@LakshmiKumar23 - exports with Python Script with target hardware might not work as expected, are all these exports required to build?

@LakshmiKumar23 - exports with Python Script with target hardware might not work as expected, are all these exports required to build?

@kiritigowda These are the instructions I got from the team. I am guessing it's needed

CuPy install added to develop - PR #1050

@LakshmiKumar23 open a new issue to add CuPy to dockers, will merge develop to test.