万丈高楼平地起~
每次都会因为环境的问题纠结许久,故写个教程,避免来回折腾;
本人的环境为ubuntu20.04;
docker的版本为20.10.21 ,docker版本需要大于19;
![](https://i-blog.csdnimg.cn/blog_migrate/ef70deed45513ce78c7e1f08aa0c0c1e.png)
推荐一个集成GPU的dockerhub链接,如下:
https://hub.docker.com/r/anibali/pytorch/tags?page=1&ordering=last_updated
选择需要的镜像即可,本人选择的是:anibali/pytorch:1.10.0-cuda11.3-ubuntu20.04
![](https://i-blog.csdnimg.cn/blog_migrate/c99ef472c99085f709048cd1395849ff.png)
可以把镜像pull下来,节省时间;
sudo docker run -it --gpus all --network=host --shm-size=32G --name my_gpu_env -v /data:/data -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all anibali/pytorch:1.10.0-cuda11.3-ubuntu20.04
中间尝试减掉--gpus选项,结果是cpu版本的pytorch;
进入环境后,查看python的版本:
![](https://i-blog.csdnimg.cn/blog_migrate/30b068889ccf462881be9e12ccdf95a0.png)
查看相应的torch版本,以及是否支持GPU;
![](https://i-blog.csdnimg.cn/blog_migrate/7ed3240ff6645a57da9fabebb901c87f.png)
顺便贴上小编的nvidia环境;
![](https://i-blog.csdnimg.cn/blog_migrate/0ca3db72c2545ad0574c67cf79cb8a47.png)
水平有限,有问题欢迎随时交流;