> 文章列表 > 【环境配置】Windows系统下搭建Pytorch框架

【环境配置】Windows系统下搭建Pytorch框架

【环境配置】Windows系统下搭建Pytorch框架

【环境配置】Windows系统下搭建Pytorch框架

在Windows Serve 2019系统下搭建Pytorch框架

目录

  • 【环境配置】Windows系统下搭建Pytorch框架
      • 1.用驱动总裁安装显卡驱动
      • 2.在cmd运行nvidia-smi
      • 3.安装cuda
      • 4.安装cudnn
      • 5.安装pytorch的命令
        • 1.首次安装
        • 2.操作失误需要重新安装
      • 6.安装torchvision
      • 7.pycharm测试

操作系统:Windows Serve 2019
Driver Version:474.04
CUDA Version:11.4

1.用驱动总裁安装显卡驱动

2.在cmd运行nvidia-smi

CUDA版本为11.4

3.安装cuda

参考视频:https://www.bil