win10 上安装 pytorch + cuda

Posted 2022-12-19 车斗

PyTorch 学习: 第1天: win10 上安装 pytorch + cuda

2022-09-15 by Zhang

https://pytorch.org/tutorials/

关于 PyTorch

PyTorch具备2个特性，使得它与深度学习关联紧密。首先，它使用GPU加速计算，通常比在CPU上执行相同的计算速度快50倍。其次，PyTorch提供了支持通用数学表达式数值优化的工具，该工具用于训练深度学习模型。请注意，这2个特性适用于一般的科学计算，而不只适用于深度学习。事实上，我们完全可以将PyTorch描述为一个在Python中为科学计算提供优化支持的高性能库。

首选条件

Win10 Home
Anaconda3
Python 3.9
VS2017
C盘要足够大

首先桌面右键选： “NVIDIA控制面板”

1. 首选图形处理器=高性能NVIDIA处理器
2. 【帮助】->【系统信息】->【组件】查看 NVCUDA64.DLL。记下：NIVIDIA CUDA 11.5.56 driver

安装 cuda 驱动和开发包

1. cuda_11.5.2_496.13_windows.exe
   地址：https://developer.nvidia.com/cuda-toolkit-archive
   下载：CUDA Toolkit 11.5 Update 2 Downloads

2. cudnn_8.3.3.40_windows.exe
   地址：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
   下载：Download cuDNN v8.3.3 (March 18th, 2022), for CUDA 11.5

上面的软件包双击安装，路径一定选默认的。C盘要足够大。

安装 pytorch

下面的包在：https://download.pytorch.org/whl/torch_stable.html
根据上面的cuda版本选择：
3) torch-1.11.0+cu115-cp39-cp39-win_amd64.whl
4) torchaudio-0.11.0+cu115-cp39-cp39-win_amd64.whl
5) torchvision-0.12.0+cu115-cp39-cp39-win_amd64.whl

上面的包放在：C:\\DEVPACK\\Anaconda3\\downloads 目录里。

现在我们要建一个Python的虚拟环境来安装pytorch，打开cmd命令行终端或【开始】-> 【anaconda prompt】，输入：

(base) C:\\DEVPACK\\Anaconda3\\downloads > conda create -n pytorch python=3.9
(base) C:\\DEVPACK\\Anaconda3\\downloads > activate pytorch

切换至上面 whl 所在目录，执行安装：

(pytorch) C:\\DEVPACK\\Anaconda3\\downloads> pip install torch-1.11.0+cu115-cp39-cp39-win_amd64.whl
(pytorch) C:\\DEVPACK\\Anaconda3\\downloads> pip install torchvision-0.12.0+cu115-cp39-cp39-win_amd64.whl
(pytorch) C:\\DEVPACK\\Anaconda3\\downloads> pip install torchaudio-0.11.0+cu115-cp39-cp39-win_amd64.whl
(pytorch) C:\\DEVPACK\\Anaconda3\\downloads>python

Python 3.9.13 (main, Aug 25 2022, 23:51:50) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>

经过上面的步骤，pytorch with cuda support 安装成功了。上面需要的包也可以从我的百度网盘下载：

链接：https://pan.baidu.com/s/1YY1TG0cM3gx8zBp3oqh-1g 
提取码：5qeu