在做图片迁移的实验时候,
https://github.com/6o6o/chainer-fast-neuralstyle
需要安装chainer环境,在安装完成之后,跑代码可以选择用GPU跑,即在命令的最后使用 -g 0(显卡为0)。
但是发现了报错:
CUDA enviroment is not correactly set up.
排查原因是因为没有安装好cupy-cuda的版本,安装的是简单的cupy,可以用下面的命令看自己的chainer安装完成之后,是否能够支持GPU的跑动:
python
import chainer
chainer.cuda.available
如图所示,输出是false,说明目前确实是不支持GPU版本;是因为我的Chainer虽然安装了,但是没有安装CUPY的GPU版本。
先查看自己的CUDA版本
python
import torch
print(torch,cuda.version)
输出10.0.130,需要找到相关的cupy