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
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
export DEBIAN_FRONTEND=noninteractive
apt-get update
apt-get upgrade --no-install-recommends -y
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
apt-get install -y ./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_5.4.50403-1_all.deb
apt-get update
apt-get install --no-install-recommends -y \
@ -29,13 +28,14 @@ From: ubuntu:20.04
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
###########################################################################
apt-get install --no-install-recommends -y software-properties-common
add-apt-repository ppa:openkim/latest -y
apt-get update
apt-get install --no-install-recommends -y \

View File

@ -31,9 +31,9 @@ From: ubuntu:20.04
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