-
这是为PyCharm提供的PyTorch Docker仓库:pytorch-docker-for-pycharm
资源介绍
pytorch-docker-for-pycharm
这是pytorm的pytorch docker-composit仓库
安装
安装步骤1:Docker
安装步骤2:nvidia工具包(用于gpu)
步骤2-1
sudo apt-get install -y apt-transport-https ca-certificates curl gnupg-agent software-properties-common
步骤2-2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - && curl -s -L https://nvidia.github.io/nvid