Add missing hipfft-dev to ROCm containers

This commit is contained in:
Richard Berger
2023-04-06 21:14:19 -06:00
parent ebb8eee941
commit 3950765e8f
2 changed files with 9 additions and 9 deletions

View File

@ -3,20 +3,19 @@ From: ubuntu:20.04
%environment %environment
export PATH=/usr/lib/ccache:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64 export PATH=/usr/lib/ccache:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib:/opt/rocm-5.1.3/llvm/lib export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib:/opt/rocm-5.4.3/llvm/lib
%post %post
export DEBIAN_FRONTEND=noninteractive export DEBIAN_FRONTEND=noninteractive
apt-get update apt-get update
apt-get upgrade --no-install-recommends -y apt-get upgrade --no-install-recommends -y
apt-get install -y --no-install-recommends curl wget libnuma-dev gnupg ca-certificates apt-get install -y --no-install-recommends curl wget libnuma-dev gnupg ca-certificates
apt-get install --no-install-recommends -y software-properties-common
########################################################################### ###########################################################################
# ROCm 5.1.3 # ROCm 5.4.3
########################################################################### ###########################################################################
wget https://repo.radeon.com/amdgpu-install/22.10.3/ubuntu/focal/amdgpu-install_22.10.3.50103-1_all.deb wget https://repo.radeon.com/amdgpu-install/5.4.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
apt-get install -y ./amdgpu-install_22.10.3.50103-1_all.deb apt-get install -y ./amdgpu-install_5.4.50403-1_all.deb
apt-get update apt-get update
apt-get install --no-install-recommends -y \ apt-get install --no-install-recommends -y \
@ -29,13 +28,14 @@ From: ubuntu:20.04
amdgpu-install --usecase=rocm --no-dkms -y amdgpu-install --usecase=rocm --no-dkms -y
########################################################################### ###########################################################################
# ROCm hipCUB # ROCm hipCUB and hipFFT
########################################################################### ###########################################################################
apt-get install -y hipcub-dev apt-get install -y hipcub-dev hipfft-dev
########################################################################### ###########################################################################
# Common Software # Common Software
########################################################################### ###########################################################################
apt-get install --no-install-recommends -y software-properties-common
add-apt-repository ppa:openkim/latest -y add-apt-repository ppa:openkim/latest -y
apt-get update apt-get update
apt-get install --no-install-recommends -y \ apt-get install --no-install-recommends -y \

View File

@ -31,9 +31,9 @@ From: ubuntu:20.04
amdgpu-install --usecase=rocm --no-dkms -y amdgpu-install --usecase=rocm --no-dkms -y
########################################################################### ###########################################################################
# ROCm hipCUB # ROCm hipCUB and hipFFT
########################################################################### ###########################################################################
apt-get install -y hipcub-dev apt-get install -y hipcub-dev hipfft-dev
########################################################################### ###########################################################################
# Common Software # Common Software