目前实验室所使用的环境是服务器的环境,所以登进某个IP后会发现有多张卡,可是我们自己智能使用某一块具体的卡,如何指定卡呢?方法是在所使用的代码前面加上如下两行代码:
import os
**os.environ[“CUDA_VISIBLE_DEVICES”] = “gpu的物理id”()**千万要注意的是方括号里的字母不要拼错,拼错了代码是没用的,但是也不会报错!!!!!!比如题主之前就把VISIBLE拼写成了VISBLE,然后因为0号卡别人已经占满了,报了一下午的显存溢出的错误!!!
如何在pytorch代码里指定所使用的GPU是具体哪一块
最新推荐文章于 2022-11-11 15:48:41 发布