pythonpytorchanacondaconda

What is the currently recommended way to install Pytorch with CUDA enabled using conda?


Pytorch official website used to have an installation option using conda (see printscreen in this answer: https://stackoverflow.com/a/51229368/1273751)

But currently no conda option is available: printscreen pytorch website

Nevertheless, there still is pytorch on conda-forge: https://anaconda.org/conda-forge/pytorch

Is there, and what is a currently recommended way to install pytorch using conda with CUDA enabled?


Solution

  • conda install pytorch-gpu -c conda-forge 
    

    will give you the cuda enabled version of pytorch.

    If this is "recommended" is up to you to decide (The question is by whom should it be recommended?). The pytorch package on conda-forge is a community effort, not officially maintained by the pytorch developers, who have deprecated their official conda channel.