Commit f42c3bb0 authored by songlinfeng's avatar songlinfeng 💬
Browse files

Update README.md

parent deb763b7
Pipeline #3387 canceled with stages
# cupy
该仓库克隆自官方仓库,并整合了依赖的第三方依赖
#编译方法
```sh
export CUPY_INSTALL_USE_HIP=1
export ROCM_HOME=/opt/dtk
export HCC_AMDGPU_TARGET=gfx936
export CXXFLAGS="-fpermissive"
export CFLAGS="-fpermissive"
pip install wheel
pip install fastrlock==0.8.3 numpy==1.26.4
git clone http://developer.sourcefind.cn/codes/songlinfeng/cupy.git
cd cupy
python setup.py bdist_wheel
```
如果使用dtk-2604以上版本需要修改/opt/dtk/include/thrust/version.h,添加#define THRUST_DEVICE_SYSTEM THRUST_DEVICE_SYSTEM_HIP
```sh
#pragma once
#include <thrust/detail/config/config.h>
#define THRUST_DEVICE_SYSTEM THRUST_DEVICE_SYSTEM_HIP
#if THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA
#include <cuda/version>
#endif
```
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment