Anaconda虚拟环境下安装pytorch1.1

  1. 创建虚拟环境

    conda create -n $YOUR_NAME$ python=3.6
    
  2. 安装pytorch1.1.0
    conda或者pip均可,不同pytorch版本或者cuda版本,请参见:
    https://pytorch.org/get-started/previous-versions/

    pip install torch==1.1.0 torchvision==0.3.0 (pip安装,cuda、cudnn需要事先安装好)
    conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 (自动安装cuda、cudnn)
    

你可能感兴趣的:(linux软件安装,深度学习环境)