Skip to content

Latest commit

 

History

History
33 lines (23 loc) · 613 Bytes

README.md

File metadata and controls

33 lines (23 loc) · 613 Bytes
Raw

ai

install tensor RT

  • pip install nvidia-pyindex
  • pip install nvidia-tensorrt
  • pip3 install pycuda

生成文件

pip freeze >requirements.txt

compile base docker image

docker build -t guojingneo/pytorch-tensorflow-notebook ./deploy

run base docker image

docker run --gpus all -p 8888:8888 --rm -it --name pt_tf_notebook guojingneo/pytorch-tensorflow-notebook

compile app docker image

docker build -t guojingneo/ai-world .

run app docker image


docker run --gpus all -p 8888:8888 -v $(pwd):/workspace:rw --rm -it --name ai-world guojingneo/ai-world