AWS EC2 Deep Learning instance cuda 3.0 - PullRequest
Новая
       13

AWS EC2 Deep Learning instance cuda 3.0

0 голосов
/ 14 апреля 2020

Я только что запустил (и заплатил) Deep Learning AMI (Ubuntu 18.04) Version 27.0 (ami-0dbb717f493016a1a) тип экземпляра g2.2xlarge. Я активировал

for PyTorch with Python3 (CUDA 10.1 and Intel MKL) ____________source activate pytorch_p36

Когда я запускаю свою сеть Pytorch, я вижу предупреждение 

/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/cuda/__init__.py:134: UserWarning: 
    Found GPU0 GRID K520 which is of cuda capability 3.0.
    PyTorch no longer supports this GPU because it is too old.
    The minimum cuda capability that we support is 3.5.

Это реально?

Это мой код поставить свой нейронный net на GPU 

if torch.cuda.is_available():
        device = torch.device("cuda:0")  # you can continue going on here, like cuda:1 cuda:2....etc. 
        print("Running on the GPU")
    else:
        device = torch.device("cpu")
        print("Running on the CPU")

    net = Net(image_height, image_width)
    net.to(device)

1 Ответ

0 голосов
/ 17 апреля 2020

Мне пришлось использовать g3s.xlarge экземпляр. Я предполагаю, что экземпляры g2 используют старые графические процессоры.

Кроме того, я должен был сделать num_workers=0 на моих загрузчиках данных после этого https://discuss.pytorch.org/t/oserror-errno-12-cannot-allocate-memory-but-memory-usage-is-actually-normal/56027.

И это еще один гадкий питч { ссылка } при добавлении тензоров к устройству.

...