是否有一种方法可以在特定(GPU)设备的上下文中运行pytorch (而不必为每个新的张量(如.to选项)指定设备)?类似于tensorflow with tf.device('/device:GPU:0'):..。默认设备似乎是cpu (除非我做错了): a = torch.zeros(1)
>>> cpu
我试着用我的GPU使用py手电筒,但是我总是得到错误的AssertionError: Torch not compiled with CUDA enabled。)Is CUDA available: FalseGPU1.11.0 py3.9_cpu_0 pytorch
[conda] pytorch-mutex