This commit is contained in:
Родитель
76e00b2316
Коммит
8c41bb9c20
|
@ -173,6 +173,8 @@ ADD third_party third_party
|
||||||
|
|
||||||
RUN make RCCL_HOME=/opt/rccl/build/ ROCBLAS_BRANCH=release-staging/rocm-rel-6.0 ROCM_VER=rocm-5.5.0 -C third_party rocm -o cpu_hpl -o cpu_stream
|
RUN make RCCL_HOME=/opt/rccl/build/ ROCBLAS_BRANCH=release-staging/rocm-rel-6.0 ROCM_VER=rocm-5.5.0 -C third_party rocm -o cpu_hpl -o cpu_stream
|
||||||
|
|
||||||
|
ENV CXX="/opt/rocm/bin/hipcc"
|
||||||
|
|
||||||
ADD . .
|
ADD . .
|
||||||
RUN apt install rocm-cmake -y && \
|
RUN apt install rocm-cmake -y && \
|
||||||
python3 -m pip install --upgrade wheel setuptools==65.7 && \
|
python3 -m pip install --upgrade wheel setuptools==65.7 && \
|
||||||
|
|
6
setup.py
6
setup.py
|
@ -183,7 +183,7 @@ setup(
|
||||||
**x,
|
**x,
|
||||||
'develop': x['dev'] + x['test'],
|
'develop': x['dev'] + x['test'],
|
||||||
'cpuworker': x['torch'],
|
'cpuworker': x['torch'],
|
||||||
'amdworker': x['torch'] + x['ort'],
|
'amdworker': x['amd'] + x['ort'],
|
||||||
'nvworker': x['torch'] + x['ort'] + x['nvidia'],
|
'nvworker': x['torch'] + x['ort'] + x['nvidia'],
|
||||||
}
|
}
|
||||||
)(
|
)(
|
||||||
|
@ -216,6 +216,10 @@ setup(
|
||||||
'onnxruntime-gpu==1.10.0',
|
'onnxruntime-gpu==1.10.0',
|
||||||
],
|
],
|
||||||
'nvidia': ['py3nvml>=0.2.6'],
|
'nvidia': ['py3nvml>=0.2.6'],
|
||||||
|
'amd' : [
|
||||||
|
'torchvision>=0.8.0a0',
|
||||||
|
'transformers>=4.3.3, <4.23.0',
|
||||||
|
]
|
||||||
}
|
}
|
||||||
),
|
),
|
||||||
include_package_data=True,
|
include_package_data=True,
|
||||||
|
|
|
@ -47,7 +47,7 @@ if(EXISTS ${HIP_PATH})
|
||||||
# Search for hip in common locations
|
# Search for hip in common locations
|
||||||
list(APPEND CMAKE_PREFIX_PATH ${HIP_PATH} ${ROCM_PATH})
|
list(APPEND CMAKE_PREFIX_PATH ${HIP_PATH} ${ROCM_PATH})
|
||||||
set(CMAKE_PREFIX_PATH /opt/rocm ROCM_PATH)
|
set(CMAKE_PREFIX_PATH /opt/rocm ROCM_PATH)
|
||||||
set(CXX "${HIP_PATH}/hipcc")
|
set(CMAKE_CXX_COMPILER "${HIP_PATH}/hipcc")
|
||||||
set(CMAKE_MODULE_PATH "${HIP_PATH}/cmake" ${CMAKE_MODULE_PATH})
|
set(CMAKE_MODULE_PATH "${HIP_PATH}/cmake" ${CMAKE_MODULE_PATH})
|
||||||
set(CMAKE_MODULE_PATH "${HIP_PATH}/lib/cmake/hip" ${CMAKE_MODULE_PATH})
|
set(CMAKE_MODULE_PATH "${HIP_PATH}/lib/cmake/hip" ${CMAKE_MODULE_PATH})
|
||||||
endif()
|
endif()
|
||||||
|
|
Загрузка…
Ссылка в новой задаче