Advertisement

Pytorch源码安装(问题解决方案)

阅读量:

源码安装

通常情况下,源码的安装过程会借助Anaconda来完成,令人感到奇怪的是,按照官方网站所提供的常规方法进行操作:

复制代码
    export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root directory]
    
    # Install basic dependencies
    conda install numpy pyyaml mkl setuptools cmake gcc cffi
    
    # Add LAPACK support for the GPU
    conda install -c soumith magma-cuda80 # or magma-cuda75 if CUDA 7.5
    
      
      
      
      
      
      
      
    

接下来执行git clone命令,获取pytorch的源代码

复制代码
    git clone --recursive https://github.com/pytorch/pytorch
    
      
    

进入pytorch目录后,执行安装操作即可

复制代码
    python set

全部评论 (0)

还没有任何评论哟~