weixin_46858750
*蘑菇*
2021-04-02 22:25
采纳率: 100%
浏览 38

Liunx系统中,如何在conda 虚拟环境下安装cupy

Liunx系统中,如何在conda 虚拟环境下安装cupy

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • wjinjie
    AI 菌 2021-04-03 16:13
    已采纳

    在conda 虚拟环境下安装cupy,对cuda版本是由要求的,不同的cuda版本应该安装对应的cupy版本,这样能减少出错。具体的各版本安装命令如下:

    # For CUDA 8.0
    pip install cupy-cuda80
     
    # For CUDA 9.0
    pip install cupy-cuda90
     
    # For CUDA 9.1
    pip install cupy-cuda91
     
    # For CUDA 9.2
    pip install cupy-cuda92
     
    # For CUDA 10.0
    pip install cupy-cuda100
     
    # For CUDA 10.1
    pip install cupy-cuda101
     
    # Install CuPy from source
    pip install cupy
    点赞 评论

相关推荐