rocm torch tools

set_venv.sh

#! /bin/bash

export torch_system=`pip3 freeze | grep torch==`
export torchvision_system=`pip3 freeze | grep torchvision==`

virtualenv --system-site-packages ./venv

source ./venv/bin/activate

alias pip=pip3.6
alias python=python3.6

pip install $torch_system
pip install $torchvision_system

# 포기하면 편함
# freeze 하지 맙시다
# sed -i 's/include-system-site-packages \= true/include-system-site-packages = false/' ./venv/pyvenv.cfg

torch_down.sh

docker stop torch_cont

torch_shell.sh

docker exec --workdir "/data" --interactive --tty torch_cont /bin/bash

torch_up.sh

# docker pull rocm/pytorch:rocm3.3_ubuntu16.04_py3.6_pytorch

docker run -it -d -v /mnt/l/ml:/data --privileged --rm --name=torch_cont --device=/dev/kfd --device=/dev/dri --group-add video rocm/pytorch:rocm3.3_ubuntu16.04_py3.6_pytorch