【Pytorch安装】&【深度学习框架】-- GPU、cuda、cudatookit等

正确的安装方法

参照官网！！！
如果你的cuda版本低于上面官网连接中展示的，怎么办？还是参照
官网！！！这里展示了过底的cuda版本对应的版本底的PyTorch版本的安装。

比如说，可以从这个页面搜索cuda10.1应该安装什么样版本的torch。

深度学习框架 GPU

• 图形处理器（英语：graphics processing unit，缩写：GPU），俗称显卡。
  

  
  

  是硬件。
  

  
  

• 显卡驱动：（通常指NVIDIA Diver）
  想要用这个硬件就需要一个驱动（很好理解：比如电脑要连打印机就要装打印机驱动，所以电脑要用GPU也要装驱动）。
  

  
  

CUDA

• CUDA
  CUDA是一个并行计算平台和编程模型，能够使得使用GPU进行通用计算变得简单和优雅。
  

  
  

• cudann
  这个其实就是一个专门为深度学习计算设计的软件库，里面提供了很多专门的计算函数，如卷积等。
  

  
  

  方便运算。
  

  
  

• CUDA Toolkit
  CUDA 工具包的主要包含了 CUDA-C 和 CUDA-C++ 编译器、一些科学库和实用程序库、CUDA 和 library API 的代码示例、和一些 CUDA 开发工具。
  

  
  

• nvcc
  nvcc其实就是CUDA的编译器
• nvidia-smi
  nvidia-smi全程是NVIDIA System Management Interface ，它是一个基于前面介绍过的NVIDIA Management Library(NVML)构建的命令行实用工具，旨在帮助管理和监控NVIDIA GPU设备。
  

  
  

  
  参考

Torchvision

Torchvision 包含一些常用的数据集、模型、转换函数等等。

当前版本 0.5.0 包括图片分类、语义切分、目标识别、实例分割、关键点检测、视频分类等工具

nvcc和nvidia-smi显示的CUDA版本不同？
参考

报错

• RuntimeError: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.
  这种版本问题就是由于cuda和torch的版本选择不配导致的，按照上面的方法重新安装torch、torchvision等就可以拉！

注意

新建一个conda环境安装torch更方便哦！不同的任务都可以新建一个conda，来避免由于不同任务所需包的不同版本导致冲突的问题。

一旦环境出现问题，也不至于影响所有任务。