如何仅在CPU上运行Pytorch

Question

我想在CPU上运行Pytorch。但我收到此错误

RuntimeError：模块必须在设备cuda：0（device_ids [0]）上具有其参数和缓冲区，但在设备cpu上找到其中一个参数和缓冲区：>

我设置了以下参数：

os.environ["CUDA_VISIBLE_DEVICES"]="-1"
device = torch.device("cpu")
程序在GPU上运行良好。

我想在CPU上运行Pytorch。但是我收到此错误RuntimeError：模块必须在设备cuda：0（device_ids [0]）上具有其参数和缓冲区，但是在设备上发现了其中一个：cpu我已经设置了...

Answer 1

您可以在运行割炬代码之前通过外壳将CUDA_VISIBLE_DEVICES变量设置为空。