Hi all thanks for stopping by. I am a student trying to get familiar with the PyTorch and GPU computation for the purpose of self-learning.
I followed the official instructions from the PyTorch website to download all the torch package, and try to run the codes https://github.com/mlelarge/dataflowr/blob/master/Notebooks/01_intro_DLDIY.ipynb as included in this tutorial ipython notebook.
However, I am stuck on transferring the python codes into the version that uses by the cuda device.
I have checked my torch is Cuda available with the following code:
use_gpu = torch.cuda.is_available()
print('Using gpu: %s ' % use_gpu)
The run result returns “Using gpu: True”, which shows that I should be able to use the GPU that is installed in my laptop. However, when I try to run the other codes, such as:
inputs_try , labels_try = var_cgpu(inputs_try,use_gpu),var_cgpu(labels_try,use_gpu)
if use_gpu:
model_vgg = model_vgg.cuda()
It returned with an error:
RuntimeError Traceback (most recent call last)
in
----> 1 inputs_try , labels_try = var_cgpu(inputs_try,use_gpu),var_cgpu(labels_try,use_gpu)
2
3 if use_gpu:
4 model_vgg = model_vgg.cuda()
~\dataflowr\Notebooks\utils.py in var_cgpu(x, use_gpu)
45 def var_cgpu(x,use_gpu):
46 if use_gpu:
—> 47 return x.cuda()
48 else:
49 return x
~\AppData\Local\Continuum\anaconda3\envs\dataflowr\lib\site-packages\torch\cuda_init_.py in _lazy_init()
161 "Cannot re-initialize CUDA in forked subprocess. " + msg)
162 _check_driver()
→ 163 torch._C._cuda_init()
164 _cudart = _load_cudart()
165 _cudart.cudaGetErrorName.restype = ctypes.c_char_p
RuntimeError: CUDA error: unknown error
You may click on this link to view the screen capture of this error https://ibb.co/pbMD7kt.
My System details are as follow:
It would be really really great if someone could give me an insight in how to solve this problem. I have tried many times of uninstalling the Pytorch, Cuda tools and the graphic drivers, but the problem still exists.
Thanks in advance guys.