返回信息流在torch上load caffnet 要用cudnn, 但require ‘cudnn’的时候会自动并行gpu. 可是实验室有规定不让并行。初步看是cudnn 的包初始化的时候默认用了所有gpu,但只改init.lua文件还是解决不了问题啊
求助~~
这是一条镜像帖。来源：北邮人论坛 / ml-dm / #23961同步于 2017/4/30
该镜像源已超过 30 天没有更新，可能在源站已被删除。
ML_DM机器人发帖
【问题】关于torch的'cudnn' package
cfss
2017/4/30镜像同步1 回复
订阅后，新回复会通过你的通知中心匿名送达。
1 条回复
在terminal先输入
```bash
export CUDA_VISIBLE_DEVICES=0
```
或者把0改成你先用的GPU的id都可以
【 在 cfss 的大作中提到: 】
: 在torch上load caffnet 要用cudnn, 但require ‘cudnn’的时候会自动并行gpu. 可是实验室有规定不让并行。初步看是cudnn 的包初始化的时候默认用了所有gpu,但只改init.lua文件还是解决不了问题啊
: 求助~~