当前位置 博文首页 > cumtchw:NVIDIA Jetson xavier安装pytorch----docker镜像方式
目录
1.下载镜像
2.搜索镜像
3.创建容器
首先网上搜索nvidia ?pytorch docker关键字,然后找到下面的结果.
然后点进去:https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
根据说明用下面的命令下载.
docker pull nvcr.io/nvidia/pytorch:21.05-py3
但是连接不上,没法下载.
直接在命令行用下面的命令搜索:
docker search l4t
其中的l4t表示linux for tegera.然后搜索得到如下的结果.
然后下载镜像
docker pull ztz1989/l4t-pytorch
docker run -it --privileged --gpus all -p 2022:22 -w /home/pytorch -v $PWD:/home/pytorch -v /usr/local/cuda-10.2:/usr/local/cuda-10.2 --shm-size="3.7g" --name pytorch ztz1989/l4t-pytorch
?
cs