Pre Required :ubuntu 22.04python 3.10.6 Reference :https://rocm.docs.amd.com/en/latest/how_to/tensorflow_install/tensorflow_install.html#../../deploy/linux/install.md Install python pip and rocm package : sudo apt install python3-pipsudo apt install rocm-libs rccl Use Virtual Environment: sudo pip3 install virtualenvvirtualenv psi-venvsource psi-venv/bin/activateInstall tensor flow, tesorflow-rocm and protobuf pip install tensorflow=2.12pip install tensorflow-rocm==2.12.0.560 –upgradepip install protobuf==3.20.3 Verify Python is Running :python3 -c ‘import tensorflow’ 2> /dev/null && […]
Tag: MI100
AMD GPU Benchmark
Ubuntu 22.04 Install docker : https://www.digitalocean.com/community/tutorials/how-to-install-and-use-docker-on-ubuntu-22-04 Install AMD TF-ROCm Image https://rocm.docs.amd.com/en/latest/how_to/tensorflow_install/tensorflow_install.html#../../deploy/linux/install.md docker pull rocm/tensorflow:latest Run it and enter the docker console docker run -it –network=host –device=/dev/kfd –device=/dev/dri \ –ipc=host –shm-size 16G –group-add video –cap-add=SYS_PTRACE \ –security-opt seccomp=unconfined rocm/tensorflow:latest Do benchmarking total images/sec: 477.72