docker-rocm3.7.sh 767 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
WORKSPACE=$1
echo "workspace: " $WORKSPACE

docker run                                                                   \
-it                                                                          \
--rm                                                                         \
--privileged                                                                 \
--group-add sudo                                                             \
-w /root/workspace                                                           \
-v $WORKSPACE:/root/workspace                                                \
asroy/tensorflow:rocm3.7-tf2.3-dev-omp                                       \
/bin/bash

#--network host                                                               \