关于使用下载安装cupy库的一些经验

这是在anaconda虚拟环境中下载安装cupy库的一些经验.

1. 先下载安装visual studio, 安装其中的C++桌面开发就好, 记好visual studio下载到那个文件夹了.

2. 不要想着使用最新的cuda, cupy库有可能不支持某些版本的cuda! 详情访问如下网站

https://docs.cupy.dev/en/stable/install.html

我下载的是11.8版本的.

3. 下载cudnn时登录缓慢时, 可以把登录网站的后缀从.com改为.cn, 这样至少比使用.com登录快. 下载的cudnn需要与对应的cuda匹配.

4. 注意需要将cuda的一些文件位置加到PATH中.

5. 注意需要把visual studio的cl.exe所在的目录添加到PATH中, 否则cupy可能仍然无法运行. 可能在

C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.37.32822\bin\Hostx64\x64

6. 正式下载安装cupy

在Anaconda Prompt中激活虚拟环境

conda activate xxxx

下载cupy库, 我的是11.8版本的, 那么按如下命令下载cupy库

pip 3 install cupy-cuda11x

7. 验证cupy可否运行

import cupy as cp

x = cp.arange(6).reshape(2, 3)
print(x)

y = x.sum(axis=1)
print(y)

在少量数据的情况下, cupy运行得比numpy慢, 因为数据读取到gpu需要很多的时间.