• win10 上安装 pytorch + cuda


    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
    
    • 1
    • 2

    切换至上面 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
    >>>
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11

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

    链接:https://pan.baidu.com/s/1YY1TG0cM3gx8zBp3oqh-1g 
    提取码:5qeu 
    
    • 1
    • 2
  • 相关阅读:
    【Java】Integer包装类底层设计原理
    笔记:在Entity Framework Core中使用乐观并发控制来处理数据更新的冲突
    Elasticsearch7 添加密码验证、并且使用postman访问带密码的es
    使用 Docker 部署 VS Code in The Browser
    如何在mac上安装多版本python并配置PATH
    http 知识整理
    U2-Net显著性检测源码详解
    01 TextRNN & FastText & TextCNN-04-训练要点,实验过程
    前端学习笔记
    Linux是什么,其特点是啥
  • 原文地址:https://blog.csdn.net/cheungmine/article/details/126875726