mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
75a2d8e2de
Summary: Drop custom hcc/hip as the 1.9.2 release should contain the relevant patches therein. Most notable feature in 1.9.2 is mixed precision support in rocBLAS and MIOpen. These features will be enabled by subsequent PRs. bddppq ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/14216 Differential Revision: D13354294 Pulled By: bddppq fbshipit-source-id: 2541d4a196af21c9432c1aff7f6e65b572628028 |
||
|---|---|---|
| .. | ||
| jenkins | ||
| 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-cuda8-cudnn6-all-options | ||
| ubuntu-16.04-cuda8-cudnn7-all-options | ||
| ubuntu-16.04-gpu-tutorial | ||
| 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