http://pytorch.org/
La méthode d'installation est décrite d'une manière facile à comprendre, mais elle est décrite comme un mémorandum.
Tout d'abord, créez un environnement avec "Faire fonctionner Keras 2.0 et OpenCV 3.2 dans un environnement GPU".
pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl
pip install torchvision
La commande suivante apprendra les données MNIST dans l'environnement GPU.
curl -kL https://raw.githubusercontent.com/pytorch/examples/master/mnist/main.py | python
Recommended Posts