GPU版本PyTorch详细安装教程

这篇具有很好参考价值的文章主要介绍了GPU版本PyTorch详细安装教程。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

目录

一、安装显卡驱动

1、查看显卡驱动型号

2、下载显卡驱动

3、查看GPU状态

二、安装Visual Studio 2019

三、安装CUDA

1、下载对应版本的CUDA

2、安装下载好的CUDA

3、设置环境变量

 四、安装cudnn

五、安装anaconda

六、安装PyTorch

1、创建虚拟环境

2、激活并进入虚拟环境

3、安装PyTorch

4、验证PyTorch是否安装成功

注意:30系列的的显卡暂时不支持cuda11以下版本!!!

一、安装显卡驱动

1、查看显卡驱动型号

右击右下角开始,在设备管理器中查看计算机显卡型号,例如我的显卡是GTX1050:

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

2、下载显卡驱动

进入英伟达官网,下载对应的显卡驱动:官方驱动 | NVIDIAhttps://www.nvidia.cn/Download/index.aspx?lang=cn

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

选择对应版本参数并下载安装,安装时直接一直下一步即可。

3、查看GPU状态

安装完成后,打开CMD终端(快捷键:Win+R),命令行输入:nvidia-smi查看GPU状态:

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

 注意:右上角显示的CUDA版本(红色框中)是当前版本所支持的最高CUDA版本。

二、安装Visual Studio 2019

由于要使用cuda则必须要有一个编译工具,这里安装的是Visual Studio 2019

下载Visual Studio 社区版链接https://visualstudio.microsoft.com/zh-hans/downloads/

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

安装完成后进入以下界面,注意:勾选“Python开发“C++桌面开发

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

三、安装CUDA

1、下载对应版本的CUDA

 CUDA各版本官方下载地址: CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive

 CUDA用的是10.2版本,因此进入cuda官网找到对应的版本进行下载:

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

 pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

2、安装下载好的CUDA

下载完成后将CUDA安装,注意:建议默认路径,后期需要添加环境变量,安装时选择自定义安装

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

勾选Visual Studio Integration:

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

3、设置环境变量

鼠标右键计算机(此电脑),打开属性->高级系统设置->环境变量,可以看到系统变量中多了CUDA_PATHCUDA_PATH_V10_2两个环境变量。

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

接下来,还要在系统变量中“新建”,添加以下几个环境变量,以下是默认安装位置的路径的环境变量(左边是变量名,右边是变量值):

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2

CUDA_LIB_PATH = %CUDA_PATH%\lib\x64

CUDA_BIN_PATH = %CUDA_PATH%\bin

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

在系统变量 Path 的末尾添加:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;

双击Path,再添加如下5(默认安装路径):C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

 pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

四、安装cudnn

1、下载cuda10.2对应的cudnn

       cudnn下载地址https://developer.nvidia.com/cudnn

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

2、下载cudnn后直接将其解开压缩包,然后需要将解压后的bin,include,lib文件夹复制粘贴到cuda安装时的默认路径文件夹下(比如我的是在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 下)

注意:对整个文件夹bin,include,lib复制粘贴

3、最后测试cuda是否配置成功:

Win+R打开CMD终端执行:nvcc -V 即可看到cuda的信息:

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

五、安装anaconda

详细安装教程可参考以下链接:

(42条消息) Python安装教程步骤1:Pycharm和Anaconda的安装_ZHW_AI课题组的博客-CSDN博客https://blog.csdn.net/m0_37758063/article/details/116540837?spm=1001.2014.3001.5502

六、安装PyTorch

1、创建虚拟环境

在anaconda中添加镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

在开始栏打开刚才安装的anaconda prompt 创建项目运行虚拟环境:

conda create -n your_env_name(虚拟环境名称) python==xx(想要创建的虚拟环境的python版本号)

注:若不知道python版本,Win+R打开CMD终端执行:python,然后enter即可看到所安装的python版本号。

可以根据自己需要修改,例如:conda create -n mytorch python==3.9.7

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

输入y,然后按下enter确认开始下载安装。

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

出现以上界面则虚拟环境已经创建完成。

2、激活并进入虚拟环境

1、在上一步的基础上,对创建完成的虚拟环境进行激活,在anaconda prompt 终端中输入以下指令:

conda activate mytorch

2、若要退出当前的虚拟环境,执行以下指令即可:

conda deactivate

3、安装PyTorch

1、此时我们已经激活并进入到所创建的mytorch环境中。 

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

2、接下来进入PyTorch官网,选择相关参数,获取PyTorch安装指令,并在anaconda prompt 终端中执行红色框中的指令(该指令为最新版本的PyTorch)

注意:安装的时候要将命令后的-c pytorch后面的内容删除,从国内源进行下载,速度快一些。PyTorchhttps://pytorch.org/pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

另:若需要选择其他版本的PyTorch点击绿色框中的指令即可 

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

然后进入以下页面,找到需要的版本进行安装即可

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

3、输入PyTorch安装指令指令后,查看conda找到将要安装的包是否是要进行安装的

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

如果是,输入y之后,按下enter确认进行下载;

如果不是,就需要重新检查安装PyTorch的命令。

3、 安装PyTorch,至此,基础环境已经部署完成。

4、验证PyTorch是否安装成功

1、安装完成后,继续在在anaconda prompt 终端中执行以下指令,验证PyTorch是否安装成功

python 
import torch 
torch.cuda.is_available() 

2、如下图提示True说明框架配置成功,且GPU可用

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

 3、验证完成后Ctrl+Z回到命令行,然后执行conda list指令就可以看到该虚拟环境下已经安装好的包

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

pytorch(gpu版)详细安装教程,嵌入式深度学习开发,pytorch,深度学习,人工智能

能够在conda list中找到以上两个包,则说明已经配置完成。文章来源地址https://www.toymoban.com/news/detail-790822.html

到了这里,关于GPU版本PyTorch详细安装教程的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装

    🌷🍁 博主 libin9iOak带您 Go to New World.✨🍁 🦄 个人主页——libin9iOak的博客🎐 🐳 《面试题大全》 文章图文并茂🦕生动形象🦖简单易学!欢迎大家来踩踩~🌺 🌊 《IDEA开发秘籍》学会IDEA常用操作,工作效率翻倍~💐 🪁🍁 希望本文能够给您带来一定的帮助🌸文章粗浅,敬

    2024年02月12日
    浏览(31)
  • 如何用conda安装PyTorch(windows、GPU)最全安装教程(cudatoolkit、python、PyTorch、Anaconda版本对应问题)(完美解决安装CPU而不是GPU的问题)

            安装PyTorch的开发环境:Anaconda+CUDA+cuDNN+PyCharm Community 1.1 版本选择 第一步就是最关键的版本对应问题(这决定你能否成功安装PyTorch,以及能否成功安装GPU版本的关键问题),可以这么说,版本不能对应好,后面有很大的问题,因此,我们要先确定版本的对应关系。(

    2024年02月07日
    浏览(35)
  • 全网最稳妥通用的GPU版本Pytorch安装教程(Anaconda虚拟环境)

    朋友们,你们有没有为安装GPU版本的pytorch而苦恼过?搜的明明是GPU的安装教程,一顿操作猛如虎,最后print(torch.cuda.is_available())结果居然是False。不要慌,接下来给你介绍一个在anaconda虚拟环境下安装的最稳妥的教程!!! CUDA 是一个运行平台,是用于“GPU通用计算” cudnn 深

    2024年02月07日
    浏览(216)
  • pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南

    🌷🍁 博主 libin9iOak带您 Go to New World.✨🍁 🦄 个人主页——libin9iOak的博客🎐 🐳 《面试题大全》 文章图文并茂🦕生动形象🦖简单易学!欢迎大家来踩踩~🌺 🌊 《IDEA开发秘籍》学会IDEA常用操作,工作效率翻倍~💐 🪁🍁 希望本文能够给您带来一定的帮助🌸文章粗浅,敬

    2024年02月16日
    浏览(57)
  • 集显安装pytorch教程,cuda和cudann环境配置torch-GPU版本看这一篇就够了

    集显看最下面 首先GPU安装教程 1.安装Anaconda (这个是可以安装环境和前置的软件),这个网址是清华源下载,可以去官网下载 Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 2.安装CUDA (这个是显卡驱动,需要安装好以用显卡计算,缩短运行时间)(c

    2024年02月02日
    浏览(53)
  • 解决conda安装的pytorch是CPU版本/conda安装pytorch的GPU版本

    激活conda环境,并查看 发现pytorch版本是cpu_py38*********,代表是CPU版本。 (别太在意原有的库包,因为cpu版本关联了很多包,一个是cpu版本会有不少包都是cpu版本,重装后缺什么再安装什么就是了) 我这里指定3.8的py版本重装一个叫env的环境 查看当前的conda源中,pytorch的安装

    2024年02月05日
    浏览(49)
  • Windows环境下GPU版本pytorch安装

    通过官网引导就可以完成安装,或者通过下面网址复制conda安装命令,安装历史版本 Previous PyTorch Versions | PyTorch ​ 输出: 参考: Windows环境下Gpu版本的Pytorch安装_pytorch向下兼容吗_晓码bigdata的博客-CSDN博客

    2024年02月12日
    浏览(38)
  • 基于全新电脑环境安装pytorch的GPU版本

    距离第一次安装深度学习的GPU环境已经过去了4年多(当时TensorFlow特别麻烦),现在发现安装pytorch的GPU版本还是很简单方便的,流程记录如下。 Free Download | Anaconda 直接下载最新版本到电脑里,并安装。 win+R 然后输入cmd调出命令窗,输入 nvidia-smi 通过搜索找到”设备管理器”

    2024年02月10日
    浏览(22)
  • linux服务器安装gpu版本的pytorch

    每次换个环境又要安装pytorch,最开始安装总是出错,写下不踩坑安装全过程 运行命令 nvidia-smi 如下所示  比方说我的是右上角显示11.2,所以  选择Linux ,这里没有我的cuda版本    找过往版本Previous PyTorch Versions | PyTorch  发现有11.3  11.1版本和10.2版本,但是没有11.2版本,查

    2024年02月04日
    浏览(35)
  • GPU版PyTorch对应安装教程

    ①算力:需要先知道你的显卡,之后根据官网表格进行对应,得到算力 ②CUDA driver version:电脑上显卡的硬件驱动 ③CUDA runtime version:pytorch官网上所显示的CUDA版本号 三者之间需要满足的关系: CUDA driver version ≥ CUDA runtime version (即:②≥③) CUDA runtime version得支持自己电脑G

    2024年02月14日
    浏览(24)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包