You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
5 lines
97 B
5 lines
97 B
3 years ago
|
import torch
|
||
|
print(torch.cuda.is_available())
|
||
|
|
||
|
import os
|
||
|
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
|