在Windows中安装PyTorch

文章目录

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

  1. 创建虚拟环境

  具体查看我之前写的

  《在Windows中利用Python的venv和virtualenv创建虚拟环境》

  2. 检查显卡版本和CUDA

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

  3. 下载链接

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

  4. 下载

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

  5. 等待

  6. 检测

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