1、切换到指定的虚拟环境
source activate pytorch1.5
2、执行以下语句安装cupy,注意cuda102对应cuda10.2版本,同理cuda101对应10.1版本,以此类推,==后面是指定cupy版本,若无固定版本需求可不加
pip install cupy-cuda102==7.8.0
3、若不知道cuda版本,或没有安装pip,或不会创建虚拟环境,请查看我前面安装python和安装pytorch的文章
1、切换到指定的虚拟环境
source activate pytorch1.5
2、执行以下语句安装cupy,注意cuda102对应cuda10.2版本,同理cuda101对应10.1版本,以此类推,==后面是指定cupy版本,若无固定版本需求可不加
pip install cupy-cuda102==7.8.0
3、若不知道cuda版本,或没有安装pip,或不会创建虚拟环境,请查看我前面安装python和安装pytorch的文章