python 中安装cupy库。CuPy(A NumPy-compatible array library accelerated by CUDA)是NumPy兼容多维数组在CUDA上的实现,因此安装时需要根据cuda版本安装
# For CUDA 10.1
pip install cupy-cuda101
安装完成,import cupy as cp 后运行程序提示“ImportError: CuPy is not correctly installed”
在不断的卸载、安装cupy,以及各种更新相关依赖包(inda、wheel、pip等)之后,运行程序依旧不断提示“ImportError: CuPy is not correctly installed”o(╥﹏╥)o
崩溃之前发现有人设置了系统的环境变量,那么这个问题可能是环境变量的问题,于是一不做二不休把电脑和pycharm的环境变量统一修改了一下:
1)电脑环境变量设置
路径:电脑—右键属性—高级系统设置—高级—环境变量。
在用户变量中的Path变量,添加CUDA10.0安装路径到系统内
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64
2)Pycharm环境变量设置
路径:Run—Edit Configurations…—Enviroment variables
添加环境变量
再次运行,成功!