Solución:
Según el sitio web de Pytorch, puede instalar pytorch-cpu con
conda install pytorch-cpu torchvision-cpu -c pytorch
Puede ver en los archivos en la nube de Anaconda, que el tamaño varía entre 26 y 56 MB según el sistema operativo donde desee instalarlo.
Puedes conseguir la rueda de http://download.pytorch.org/whl/cpu/. La rueda tiene 87 MB.
Puede configurar la instalación poniendo el enlace a la rueda en el requirements.txt expediente. Si usa Python 3.6 en Heroku:
http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
de lo contrario, para Python 2.7:
http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
Por ejemplo, si sus requisitos son pytorch-cpu, numpy y scipy y estás usando Python 3.6, el requirements.txt se vería así:
http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
numpy
scipy
A partir de PyTorch 1.3, PyTorch ha cambiado su API. Para instalar solo la versión de la CPU, use
conda install pytorch torchvision cpuonly -c pytorch
Y, los archivos de rueda correspondientes se pueden descargar desde https://download.pytorch.org/whl/torch_stable.html y se pueden instalar usando pip o use el comando similar al siguiente correspondiente a sus versiones previstas de pytorch y torchvision
pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
Consulte la guía de introducción de PyTorch.
En 2020, utilice el siguiente comando si desea descargar la versión de pytorch-cpu con pip3 (en Linux y Win):
pip3 install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html