win10下在Anaconda中进行pytorch安装,并配置cuda

    技术2023-07-04  89

    整理整个pytorch的入门过程

    1、pytorch的官网教程 How to get started

            https://pytorch.org/get-started/locally/#anaconda

            为了充分利用PyTorch的CUDA支持的能力,建议(但并非必须)自己的Windows系统有NVIDIA GPU。

    2、安装Anaconda3(对应python3.7)

            Anaconda是一个管理环境、安装各种包的平台。可以使用可视化的界面Anaconda Navigator,也可以使用命令行的方式进行操作,命令行入口是Anaconda Prompt。如果让系统得知你在命令行想用Anaconda呢,就是特殊的命令格式:conda ....

             Anaconda的介绍、安装及使用教程连接如下。

             https://zhuanlan.zhihu.com/p/32925500

             其中,使用教程中包括了创建环境、更改环境、关闭环境。“环境”就是指开发环境,比如要用哪一个版本的python。此外,使用教程也包括了如何安装一个想用的包。

             官网可以下载Windows对应的安装包 Anaconda3-2020.02-Windows-x86_64.exe。

             安装完Anaconda后,自带了python 3.7.4,在cmd中查找不到,只有在Anaconda Prompt的环境中才能查到。

             相比较pip而言,通过Anaconda安装pytorch更容易,因为它会自动下载所有pytorch所依赖的包。在pytorch的官网,在开头就有一个安装命令提供。

    3、 配置 CUDA和CUDNN

              CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

             NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如加州大学伯克利分校的流行caffe软件。简单的,插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是调整性能,同时还可以在GPU上实现高性能现代并行计算。 ———————————————— 原文链接:https://blog.csdn.net/qq_35239859/article/details/79827203

    打开计算机->属性->控制面板->NVIDIV控制面板->系统信息,如下图所示。

     

               下述文章解释了如何安装正确版本的CUDA。不论是tensorflow还是pytorch,都要注意版本所需的cuda。

                https://blog.csdn.net/qq_38163755/article/details/88583016

                服务器的NVIDIA驱动程序版本允许安装CUDA10.1,在Anaconda Prompt3中,通过以下语句安装CUDA和CUDNN。

    conda install cudatoolkit=10.1 conda install cudnn

     4 安装pytorch         

                完成cuda的安装后,在pytorch官网https://pytorch.org/找到之前的版本,选择可以适配cuda10.1的pytorch,如下图。

    安装完成后,验证pytorch是否安装成功,用官网提供的命令

    首先在Anaconda Prompt中在命令行输入:

    python

    会跳出Python的环境,即">>>” ,then enter the following code:

    from __future__ import print_function import torch x = torch.rand(5, 3) print(x)

    结果类似于:

    tensor([[0.3380, 0.3845, 0.3217], [0.8337, 0.9050, 0.2650], [0.2979, 0.7141, 0.9069], [0.1449, 0.1132, 0.1375], [0.4675, 0.3947, 0.1426]])

    这样,就安装完成了。

            如果要查看此时安装的pytorch是否可以使用 GPU,输入以下代码测试CUDA是否有效。

    import torch torch.cuda.is_available()

     

    Processed: 0.010, SQL: 9