http://pytorch.org/
The installation method is described in an easy-to-understand manner, but it is described as a memorandum.
First, create an environment with "Make Keras 2.0 and OpenCV 3.2 work in GPU environment".
pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl
pip install torchvision
The following command will learn MNIST data in the GPU environment.
curl -kL https://raw.githubusercontent.com/pytorch/examples/master/mnist/main.py | python
Recommended Posts