Contents
  1. 1. Install ROCm
  2. 2. Install docker
  3. 3. Get ROCm image
  4. 4. Run a docker
  5. 5. Verify a docker
  6. 6. Note
  7. 7. Reference

How to setup ROCm docker(Ubuntu 18.04).

Install ROCm

ROCm Installation Guide

Install docker

Get ROCm image

  • ROCm

    1
    docker pull rocm/rocm-terminal
  • Caffe

    1
    sudo docker pull rocm/hipcaffe
  • TensorFlow

    1
    sudo docker pull rocm/tensorflow

Run a docker

1
2
3
4
5
alias drun='sudo docker run -it -v $HOME:/data --privileged --device=/dev/kfd'

drun rocm/rocm-terminal
drun rocm/hipcaffe
drun rocm/tensorflow

Verify a docker

  • rocminfo to find out a GPU

    1
    /opt/rocm/bin/rocminfo
  • run a tensorflow hello example

    1
    2
    3
    4
    5
    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow')
    sess = tf.Session()
    # show gpu support if anyG
    sess.run(hello)

Note

  • rock-dkms in native OS should work with thunk in docker

Reference

Contents
  1. 1. Install ROCm
  2. 2. Install docker
  3. 3. Get ROCm image
  4. 4. Run a docker
  5. 5. Verify a docker
  6. 6. Note
  7. 7. Reference