Ignore specific GPU with pytorch in Anaconda

[KFrank]

Hi Asmaron!

@Asmaron

how can I make sure that Torch only tries to run off GPU-0, aka the 3070 while ignoring the 1070 and it’s insufficient CUDA version

Run the process that runs your pytorch code under control of the
CUDA_VISIBLE_DEVICES environment variable. See, for example, this
thread:

What does “export CUDA_VISIBLE_DEVICES=1” really do?

In a multi-GPU computer(Ubuntu 16), I want to use GPU1 and do the following setting in the shell: $ export CUDA_VISIBLE_DEVICES=1 $ python But, I got an error when I set the device of the tensor: Python 3.8.3 (default, May 19 2020, 18:47:26) [GCC 7.3.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> a = torch.Tensor(1).to("cuda:0") >>> a = torch.Tensor(1).to("cuda:1") Traceback (most recent call last): File "<stdin>",…

For example, if you were to run your pytorch code under a simple python
session (on unix – I’m not sure how to modify this for windows), you would
run

CUDA_VISIBLE_DEVICES=0 python

to launch your python session.

Best.

K. Frank