Spaces:
Runtime error
Runtime error
File size: 621 Bytes
c5ca37a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# Set up the Environment
Pull docker from Docker Hub at: `chunyl/pytorch-transformers:v2`, and run it using the following script:
```
SCRIPTPATH="/home/chunyl/azure_mounts/optimus_azure"
IMAGE=chunyl/pytorch-transformers:v2
docker run \
--runtime=nvidia \
-it --rm \
--net host \
--volume $SCRIPTPATH:/workspace \
--interactive --tty $IMAGE /bin/bash
```
There is an example at `code/scripts/scripts_docker/run_docker.sh`. Please edit the project path to the absolute path on your computer by changing the "SCRIPTPATH", then run the docker at the the directory "code":
sh scripts/scripts_docker/run_docker.sh
|