Object Detector Env Preparation: Pytorch + Yolo + Anaconda + Pycharm

• NVIDIA Driver update/download
• Anaconda-based pytorch packages download
• Pycharm download & verify the cuda/cudnn
• 参考

NVIDIA Driver update/download

• GPU: GTX 1050
  
• Driver version
  

Anaconda-based pytorch packages download

• download from here: https://blog.csdn.net/didiaopao/article/details/119787139
• Create a env for latter useage(named after pytorch):
• Activate the env:
• Change the source(For sake of saving time):
• Check for version support(From this site https://pytorch.org/, do not add this last words(source problem)):
  

Pycharm download & verify the cuda/cudnn

• Change the env to the pytorch created former, then run the following code in pycharm:

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

参考

• https://blog.csdn.net/didiaopao/article/details/119787139