如何在pytorch代码里指定所使用的GPU是具体哪一块

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