Skip to content

Latest commit

 

History

History
 
 

docker

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Dockerfile
Dockerfile
 
 
README.md
README.md
 
 

README.md

Run the following command to build the docker for tensorcircuit at parent path:

sudo docker build . -f docker/Dockerfile -t tensorcircuit

One can also pull the official image from DockerHub as

sudo docker pull tensorcircuit/tensorcircuit

Run the docker container by the following command:

sudo docker run -it --network host --gpus all tensorcircuit

# if one also wants mount local source code, also add args `-v "$(pwd)":/app`

# for old dockerfile with no runtime env setting
# sudo docker run -it --network host -e LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/targets/x86_64-linux/lib -e PYTHONPATH=/app -v "$(pwd)":/app  --gpus all tensorcircuit

export TF_CPP_MIN_LOG_LEVEL=3 maybe necessary since jax suprisingly frequently complain about ptxas version problem. And export CUDA_VISIBLE_DEVICES=-1 if you want to test only on CPU.