Windows电脑部署pytorch环境

前置条件-先了解需要安装哪些基础工具

Pytorch 使用不同版本的 cuda - yhjoker - 博客园

1、安装CUDA Toolkit

1.1 官方下载

找到本地NVIDIA控制面板，得到 本地 CUDA 驱动的版本号

我这里是 10.2.141

然后去官网找对应的工具包版本号

 

 下载好之后点击安装

这里耗时会有点长 

安装完成后，验证是否能正常驱动  nvcc --version

2、安装 cuDNN

2.1官方下载

https://developer.nvidia.com/rdp/cudnn-download

这里需要注册登陆一下

下载后解压

复制粘贴到CUDA安装目录

测试是否安装成功，双击运行 bandwidthTest.exe

 显示 result=PASS

3、安装pytorch

3、1 使用 conda

官网   Previous PyTorch Versions | PyTorch

这时候需要去安装 Anaconda

官网    https://www.anaconda.com/products/individual

 3.2 使用 pip

3.3 新建一个python项目

选择 anaconda 新建的 pytouch 环境

 

引入 torch 打印结果

 到这一步就完成环境的安装了