Win10安装PyTorch

新手小白一个, 简单记录一下过程.

一  一些名词

1. GPU: Graphics Processing Unit, 图形处理器.

2. CUDA: Compute Unified Device Architecture, 是显卡厂商NVIDIA(英伟达)推出的运算平台.

                 CUDA是一种通用并行计算架构, 该架构使GPU能够解决复杂的计算问题.

二  查看电脑的CUDA版本

可自行百度通过控制面板查看, 本人实在是没有在控制面板下找到NVIDIA控制面板，所以提供以下方法.

1. 在桌面右击，选择NVIDIA控制面板

2. 选择“帮助”--“系统信息”--“组件”, 查看NVCUDA.DLL, 即可看到版本号.

三  通过Anaconda安装PyTorch

1. 官网https://pytorch.org/   , Get Started ,  根据自己电脑版本来选择.

 

 2. 打开Anaconda prompts输入官网给出的命令.

 

 3. 安装成功

 

 4. 查看版本

import torch
print(torch.__version__)

1.6.0