docker-rocm4.1.sh 755 Bytes
Newer Older
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                                                \
rocm/tensorflow:rocm4.1-tf1.15-dev                               \
/bin/bash

#--network host                                                               \