Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update NGC (#8652) 

* update NGC

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
This commit is contained in:
Jirka Borovec 2021-08-02 18:05:36 +02:00 committed by GitHub
parent cf0d362658
commit 66cc505339
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dockers/nvidia/Dockerfile
View File

@ -12,8 +12,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.
ARG PYTORCH_VERSION=21.07
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
FROM nvcr.io/nvidia/pytorch:21.06-py3
FROM nvcr.io/nvidia/pytorch:${PYTORCH_VERSION}-py3
LABEL maintainer="PyTorchLightning <https://github.com/PyTorchLightning>"