请问一下如何指定显卡，或者使用多卡训练

问答
1

image

jt.flags.use_cuda = 1
os.environ[‘CUDA_VISIBLE_DEVICES’]=‘1’
按照这种方式，设置了，还是不能指定显卡？

另外，请问各位大神，如何指定多卡一起训练，可能一块卡显存不够。。。。。

2

试试先设置环境变量再设置 flag？

多卡训练可以参考文档 jittor.mpi — Jittor 1.3.4.2 文档

3

补充一下，环境变量设置要放在 import 其他任何库（包括 jittor）开始之前才能生效。

4

谢谢，但是我在使用多卡时候又碰到了一个问题
x86_64-conda_cos6-linux-gnu-cc: error: unrecognized command line option ‘–showme:version’
大神知道是为什么吗

5

你先设置环境变量再设置flag成功了吗?为什么我即使是这样也不行