Commit cf86f723 authored by suily's avatar suily
Browse files

Update README.md

parent c8a60a0b
......@@ -49,7 +49,6 @@ docker build --no-cache -t vision_transformer:latest .
docker run -it --network=host --privileged=true --name=docker_name --device=/dev/kfd --device=/dev/dri --group-add video --shm-size=32G --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /path/your_code_data/:/path/your_code_data/ imageID /bin/bash
cd /your_code_path/vision_transformer
pip install -r requirements.txt
pip install -r requirements.txt # flax会强制安装某版本ai包
pip uninstall tensorflow
pip uninstall jax
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment