मैंने इस आदेश का उपयोग करके PyTorch इंस्टॉल किया है:

pip3 install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

और यह काम करने लगा। मशाल आयात करते समय, कोई त्रुटि नहीं है। मेरे लैपटॉप में एक GeForce 1060 TI है, जिसके बारे में मुझे लगा कि यह काम करेगा सीयूडीए।

यहाँ IDE (ग्रहण) में त्रुटि है:

Screenshot with code

-2
William Dwinnell 19 सितंबर 2019, 05:05

1 उत्तर

सबसे बढ़िया उत्तर

आपको कोड शामिल करना चाहिए, ग्राफ़ नहीं!

PyTorch निकालें और इसे इस आदेश के साथ फिर से स्थापित करें:

conda install pytorch -c pytorch
pip3 install torchvision

इसके बाद इसे इस आदेश से जांचने का प्रयास करें:

import torch
import torchvision
train_on_gpu = torch.cuda.is_available()

if train_on_gpu:
    print('CUDA is available, Training on GPU ...')
else:
    print('CUDA is not available!  Training on CPU ...')
1
AhmaD 28 सितंबर 2019, 14:54