企业型网站有哪些特点,ui设计是什么职业,小白学网站建设与设计书,网站栏目怎么做301定向转载请注明出处#xff1a;小锋学长生活大爆炸[xfxuezhang.cn] 指定GPU不生效问题
解释#xff1a;就是使用os.environ[CUDA_VISIBLE_DEVICES] 1后#xff0c;后面使用起来仍然是cuda0.
解决#xff1a;在最开头就使用
import os
os.environ[小锋学长生活大爆炸[xfxuezhang.cn] 指定GPU不生效问题
解释就是使用os.environ[CUDA_VISIBLE_DEVICES] 1后后面使用起来仍然是cuda0.
解决在最开头就使用
import os
os.environ[CUDA_VISIBLE_DEVICES] 0,2# xxxxxx其它代码 GPU编号不一致问题 解释torch.device(cuda:0)所选择的GPU跟nvidia-smi中的GPU编号不一样
解决指定编号方式
import os
os.environ[CUDA_DEVICE_ORDER]PCI_BUS_ID# xxxxxx其它代码
