saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/docker/caffe2
History
Jeff Daily 9e58c85d08 [ROCm] remove use of HIP_PLATFORM (#47241) 
Summary:
Fixes deprecated use of the HIP_PLATFORM env var.  This env var is no longer needed to be set explicitly.  Instead, HIP_PLATFORM is automatically detected by hipcc.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/47241

Reviewed By: mruberry

Differential Revision: D24699982

Pulled By: ngimel

fbshipit-source-id: 9cd2f32e7c0c8d662832b0cbbc2988835a45961a
2020-11-03 09:54:44 -08:00
..
jenkins [ROCm] remove use of HIP_PLATFORM (#47241) 2020-11-03 09:54:44 -08:00
ubuntu-14.04-cpu-all-options
ubuntu-14.04-cpu-minimal
ubuntu-16.04-cpu-all-options
ubuntu-16.04-cpu-minimal
ubuntu-16.04-gpu-tutorial
readme.md

readme.md

Docker & Caffe2

Note: use nvidia-docker to run all GPU builds.

To get the latest source, rerun the docker builds using the Dockerfiles.

Docker images at https://hub.docker.com/r/caffe2ai/caffe2/ are a few months old, but will be refreshed soon.

Build like: docker build -t caffe2:cuda8-cudnn6-all-options .

Run like: nvidia-docker run --rm -it caffe2:cuda8-cudnn6-all-options python -m caffe2.python.operator_test.relu_op_test

For Docker on USB related instructions you can find some help on the gh-pages branch here