pytorch的安装

  • 首先必须要安装好cuda和cudann,我安装的是如下版本的cudapytorch的安装_第1张图片
  •  需要注意cuda版本和之后的pytorch版本需要匹配pytorch的安装_第2张图片
  • cuda10.0环境安装Pytorch_先说一下,这里不是博客,只是我的笔记本-CSDN博客_cuda10.0对应的pytorch
  • # CUDA 9.0
    conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

    # CUDA 10.0
    conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

    # CPU Only
    conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

  • 由于我选择的是cuda10.0,所以安装第二个版本的Pytorch

  • Pytorch官网有以下版本,但是没有cuda10.0,因此只能用上面的来安pytorch的安装_第3张图片

  •  

你可能感兴趣的:(pytorch,深度学习,python)