在Windows中安装PyTorch

文章目录

• [1. 创建虚拟环境](#1. 创建虚拟环境)
• [2. 检查显卡版本和CUDA](#2. 检查显卡版本和CUDA)
• [3. 下载链接](#3. 下载链接)
• [4. 下载](#4. 下载)
• [5. 等待](#5. 等待)
• [6. 检测](#6. 检测)

1. 创建虚拟环境

具体查看我之前写的
《在Windows中利用Python的venv和virtualenv创建虚拟环境》

2. 检查显卡版本和CUDA

这种情况是需要电脑上有单独的英伟达的显卡、或者英伟达的显卡和集显这两种情况都是可以的。）

3. 下载链接

进入PyTorch官网选择合适自己设备的下载链接

4. 下载

进入cmd 输入上一步获得的下载链接

5. 等待

6. 检测

下载完成后，进入python环境检测torch是否可用