Sign inSign up

junbingao/deep_learning

By junbingao

Updated over 5 years ago

Image
0

42

junbingao/deep_learning repository overview

Deep learning environment

2021.3 Junbin, HUST

The docker is for linux OS with Nvidia gpu.

The docker is inherit from pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel.

Up to now, the docker is usable for cuda 10.1 or newer. It contains pytorch 1.6.0, tensorflow-gpu 1.15.0, and some tools:

  • matplotlib
  • tensorboardX

Tag summary

Content type

Image

Digest

Size

4.3 GB

Last updated

over 5 years ago

docker pull junbingao/deep_learning:cuda10.1-torch1.6-tf1.15