目录

一、下载CUDA

1、什么是CUDA

2、查看电脑支持版本号

3、下载CUDA包

1)进入下列下载位置

2)选择版本

4、安装CUDA

1)双击这个文件,然后得到下列图像

2)选择自定义安装

3)取消选项Visual Integration

4)使用默认安装位置

5)安装完成

6)配置环境变量

7)测试是否成功安装

二、下载pytorch包

1、什么是torch

2、下载torch

1)复制上述链接打开,然后下拉找到torch,点开

2)找到需要的版本

3)安装库

4)检查是否安装成功

3、torchaudio库是什么

4、下载torchaudio库

5、torchvision库是什么

6、下载torchvision


一、下载CUDA

1、什么是CUDA

        CUDA(Compute Unified Device Architecture)是由NVIDIA开发的一种并行计算平台和编程模型,用于利用NVIDIA GPU进行通用计算。它提供了一个编程接口,让开发者能够将计算任务分配到GPU上执行,以实现高性能的并行计算。CUDA可以用于各种领域的计算任务,包括科学计算、图形处理、机器学习等。

2、查看电脑支持版本号

                win+R输入cmd,打开命令提示符,输入命令nvidia-smi,即可看到电脑支持最高版本的版本号。如下图所示:

3、下载CUDA包

        1)进入下列下载位置

CUDA官网icon-default.png?t=O83Ahttps://developer.nvidia.com/cuda-toolkit-archive

        2)选择版本

                得到下列画面,此时其内有各个版本的CUDA,选择尽量低于电脑版本的,防止有过高的版本不稳定的情况。

选择好版本,点开,选择系统和版本

 下载好是下列形式的安装包。

4、安装CUDA

        1)双击这个文件,然后得到下列图像

        2)选择自定义安装

        3)取消选项Visual Integration

                如果计算机上已经安装了与CUDA版本相兼容的Visual Studio Integration文件,或者通过其他方式(如单独安装Visual Studio的插件)已经集成了CUDA支持,那么在安装CUDA时勾选“Visual Studio Integration”可能会导致冲突或重复安装,进而引发问题。

        4)使用默认安装位置

                安装位置不要动,使用默认的即可,防止出现问题

        5)安装完成

        6)配置环境变量

                1、 右击此电脑,打开属性,找到高级系统设置,打开并点击环境变量即可

                2、或者直接在搜索栏搜索环境变量

                3、查看环境变量

                        查看是否有这个环境变量,如果没有则需手动增加,增加内容即为图示内容,点击新建即可

                或者在命令提示符内输入命令set cuda查看环境变量

        7)测试是否成功安装

                在命令提示符内输入命令nvcc --version 或者nvcc -V,查看是否有cuda,如果有则安装成功,如果没有则可能版本不兼容

二、下载pytorch包

1、什么是torch

        Torch是一个开源的机器学习库,主要用于构建深度学习模型。它基于Lua编程语言,并提供了一个强大的张量计算库,可以有效地进行数值计算和矩阵操作。Torch同时也提供了丰富的神经网络模块,使得用户可以方便地构建和训练深度神经网络模型。

        Torch的核心是它的张量库,它支持多维张量的操作和计算,类似于NumPy库。张量是一种多维的数组,可以用于存储和处理各种数据类型的数据。Torch的张量库提供了丰富的函数和运算符,可以进行各种数学运算、线性代数运算和张量操作等。

2、下载torch

        进入pytorch官网,选择好需要的类别,然后将下列链接复制到浏览器打开,选择版本进行下载,因为torch所占内存太大,直接用其所给命令去命令提示符下载如果网络稍微有波动则会导致下载失败,所以直接去源网页下载,下列是官网入口:

pytorch官网icon-default.png?t=O83Ahttps://pytorch.org/

        1)复制上述链接打开,然后下拉找到torch,点开

        

        2)找到需要的版本

                可以使用网页查找命令,长按ctrl+F,然后输入需要的版本,cu表示安装的cuda版本,cp表示python版本,例如下图所示,我下载的是torch1.13.0,cuda版本可以比安装的版本低,但是不能比安装的版本高,python版本为3.10,python版本可以在命令提示符内输入命令python或者where python查看,如下图所示

                where python命令可以查看安装的python的优先级,优先级高那么就使用pip命令首选就是这个版本,如果不想优先用这个版本,可以去环境变量里手动调换一下python的先后顺序。

                python命令是进入当前python的编译环境同样输出当前使用python版本号。

        3)安装库

                上述的链接既是下载链接,建议放入迅雷下载器下载,这个下载更快,其他的也行,怎么快怎么来。

                下载完是如下的文件格式,然后打开命令提示符,使用命令pip install 文件地址,即可安装。如下图所示:

pip install C:\Users\19947\Desktop\新建文件夹 (3)\torch-1.13.0+cu117-cp310-cp310-win_amd64.whl

或者直接右击复制地址:

        4)检查是否安装成功

                命令提示符输入命令pip list即可查看

3、torchaudio库是什么

        torchaudio是一个基于PyTorch的音频处理库。它提供了一系列用于加载、转换和处理音频数据的函数和类。torchaudio支持多种音频格式,包括wav、mp3、flac等,并提供了一些基本的音频处理功能,如音频剪辑、音频格式转换等。通过torchaudio,用户可以方便地在PyTorch中进行音频数据的处理和建模。

4、下载torchaudio库

        重复上述安装torch的步骤即可,选好版本号

        点开torchaudio,然后快捷键搜索相应版本,版本号和torch一致即可

        安装方式也是同样使用 pip install 文件地址 即可

5、torchvision库是什么

        torchvision是一个在PyTorch中提供了一些用于图像处理任务的工具的库。它包含了常用的数据集加载、图像变换以及预训练模型等功能。torchvision可以大大简化图像处理任务的开发流程。

6、下载torchvision

        步骤和上述一致,注意版本号,最后得到下列即为安装成功

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐