docker-rocm4.1.sh 755 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
6
7
8
9
10
WORKSPACE=$1
echo "workspace: " $WORKSPACE

docker run                                                                   \
-it                                                                          \
--rm                                                                         \
--privileged                                                                 \
--group-add sudo                                                             \
-w /root/workspace                                                           \
-v $WORKSPACE:/root/workspace                                                \
zjing14's avatar
zjing14 committed
11
rocm/tensorflow:rocm4.1-tf1.15-dev                               \
Chao Liu's avatar
Chao Liu committed
12
13
14
/bin/bash

#--network host                                                               \