Docker配置Pytorch深度學習環境

2022-09-24 06:24:09 字數 1035 閱讀 5866

$ docker pull pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel
$ docker images

$ docker run -it -v /home/cenjw/dataset/:/home/data \

> -v /home/cenjw/kt/:/home/ktmodel --gpus=all --name=torchenv \

> pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel /bin/bash

$ docker ps -a

$ docker start id
$ docker attach 容器id
docker run -it 容器名 /bin/bash 進入容器         

exit 退出容器

ctrl + p + q 退出容器不停止

刪除容器

docker rm 容器id

docker rm -f $(docker ps -aq) // all

docker ps -a -q | xargs docker rm // all

docker start id 啟動容器

docker stop id 停止容器

docker kill id 強制停止

進入當前正在 執行的容器

docker exec -it 容器id bashshell

docker attach id

docker logs -tf --tail 10 id 顯示日誌

docker inspect id 檢視元資料

樹莓派配置Pytorch

樹莓派映象 python3的安裝,這張卡里已有3.7了 python,pip設定軟鏈結到python3,pip3上 閱讀了這篇部落格,發現沒有什麼官方提供的簡便安裝方法,要在樹莓派上編譯,很慢,選擇了 github上找到倉庫包含了可以在樹莓派執行的pytorch wheel,分別有1.1和1.4版本...

linux下pytorch環境配置

先建立python環境 其中環境名為ww1 conda create n ww1 python 3.7 進入該環境 conda activate ww1 切換到該環境 source activate ww1 退出該環境 conda deactivate pytorch 8版本和你的cuda版本以及p...

docker中執行的pytorch解決多執行緒報錯

在docker中執行的pytorch執行多執行緒訓練,報如下錯誤 unexpected bus error encountered in worker.this might be caused by insufficient shared memory shm 解決方法,啟動docker時,增加 i...