Liunx系统中,如何在conda 虚拟环境下安装cupy
在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