首页 / 安装部署 / pytroch AI机器学习环境部署,RTX3080显卡

pytroch AI机器学习环境部署,RTX3080显卡

2023-04-24 12:37:24安装部署 阅读 1412
网上关于机器学习环境部署的文章资料很多,有些可能只是一部分过程,有些则写得很长很长,遇到一些比较好的但要专业知识才能看得懂。

个人曾经看了一个80分钟环境部署视频,事实上只需要一页文档就能说清楚的,视频作者没剪辑,把所有出错排错搜索的过程也放在里边。

于是自己也做一个pytroch机器学习环境的笔记,完成了在个人电脑上用RTX3080跑机器学习的环境并跑了相关脚本,分享出来,注意是windows平台。网上有用到Conda管理环境的,这里没使用。

环境部署主要分为以下几个大步骤

1、Python安装

2、Pytroch安装

3、CUDA的安装

4、CUDNN的安装

5、环境变量与测试


1、Python安装
在本次安装中选择安装python3.8,从官网下载exe并安装。


2、Pytroch安装
一般上官网下载当前最新版即可,用官网的pip命令直接安装。



再用pip命令安装,有时候cmd看似网速快,下到一半报错了,就得重来。



也可以在官网离线下载后用安装包安装,毕竟离线下载可以用加速工具,不会太慢。
到https://download.pytorch.org/whl/torch_stable.html 离线下载,如果你的pip慢的话,
用离线有工具可能比Pip快一些,根据本机python版本选,
下载后pip install torch-xxx+cpu-cp38-cp38-win_amd64.whl。


3、CUDA的安装
确定cuda版本,cuda的版本由pytroch决定, 可在pytroch那看cuda用哪个,上边截图显示了用cuda11.7。


下载cuda11.7并安装,比较大,建议用加速工具。




4、CUDNN的安装

这里是要注册登录后才能下载cudnn的,用工具加速下载。

cudnn的安装比较奇怪,解压复制到cuda目录即可,默认安装的情况下,CUDA目录于C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
打开CUDNN的解压目录,将include里的文件复制到CUDA的安装目录下的include文件夹中,将lib里的文件复制到CUDA的安装目录下的lib\x64文件夹中,将bin里文件复制到CUDA的安装目录下的bin文件夹中。

5、环境变量与测试
CUDA_PATH 为CUDA的安装路径。(默认可能在安装CUDA时已经存在)

CUDNN_PATH 为CUDNN解压目录的路径 (可以拷贝一个到CUDA目录下,方便统一打理)

在系统环境变量中找到变量Path,将CUDA的bin目录路径和CUDNN的bin目录路径添加到其值中。(默认可能在安装CUDA的时已经存在)

安装完成后,可以打开命令行工具,输入nvcc -V命令检查CUDA是否安装正确。如果输出的结果中包含CUDA的版本号,说明安装成功。

之后就正常使用了,跑不用脚本少啥就用pip装啥。

本文《pytroch AI机器学习环境部署,RTX3080显卡》由爱思考吧 isres.com 分享,转载请注明出处。本文网址:https://www.isres.com/bushu/29.html

本站主要收集测评能够节省时间和提升效率的软件工具并分享使用与学习的过程和感受,任何情况下用户都需遵守所使用软件资源的相关协议。与《pytroch AI机器学习环境部署,RTX3080显卡》有关的本站文章及资源仅供个人学习交流使用,相关资源请下载后24小时内删除,请勿用于其它用途,因此产生的任何问题由您自行承担。

相关推荐

  • pytroch AI机器学习环境部署,RTX3080显卡

    pytroch AI机器学习环境部署,RTX3080显卡

    网上关于机器学习环境部署的文章资料很多,有些可能只是一部分过程,有些则写得很长很长,遇到一些比较好的但要专业知识才能看得懂。 个人曾经看了一个80分钟环境部署视频,事实上只需要一页文档就能说清楚的,视...

    2023-04-24 12:37:24 1412
  • conda部署与物理机不同的cuda版本,一行命令部署pytroch或TensorFlow环境

    conda部署与物理机不同的cuda版本,一行命令部署pytroch或TensorFlow环境

    多套AI机器学习框架最佳的实践方法就是使用conda来管理不同的虚拟环境,以前 用pip 喜欢手动下载手动安装,什么cuda cudnn,各种版本相互依赖,conda就专治这个。但由于习惯,在使用c...

    2024-01-10 10:55:48 2035

猜你喜欢

  • 自用实践两台电脑共用一个音响的方案

    自用实践两台电脑共用一个音响的方案

    家里有两台电脑,其中有一台不常用的也只有打游戏联机的时候才开,不想单独配一个音箱,于是必须实践两台电脑共用一个音响的方案。 本次方案就是加一条音频线,连接两台电脑,效果如下: 一般主板都有音频输入和...

    2023-09-07 09:42:20 3255
  • Centos7minimal安装playwright和firefox、chromium,终于跑起来了

    Centos7minimal安装playwright和firefox、chromium,终于跑起来了

    系统环境为CentOS Linux release 7.9.2009 (Core) 最小安装,考虑到playwright的安装需要python3.7+,本次直接选择安装python3.10...

    2023-07-18 11:19:30 2238
  • 解决conda不从清华源安装 pytorch-2.1.1-py3.9_cuda11.8_cudnn8.7.0_0.tar.bz2

    解决conda不从清华源安装 pytorch-2.1.1-py3.9_cuda11.8_cudnn8.7.0_0.tar.bz2

    故障介绍 在一次服务器不是实践中,发现以下命令运行时很久没完成,下不动。 conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2...

    2024-10-21 22:25:29 933
  • Centos kvm以及kvm gui管理工具部署

    yum install -y qeum-kvm qemu-kvm-tools virt-manager libvirt virt-install yum -y install https://dl....

    2023-04-25 12:16:26 1230
  • centos7重置root密码,解决忘记密码的问题

    centos7重置root密码,解决忘记密码的问题

    重新重启,光标在界面上第一行 按e 。 输入rw single init=/bin/bash 后,按 ctrl+x 引导系统,不要按esc。 使用passwd命令重置密码,并使用touch /.au...

    2024-11-17 16:23:45 465
  • centos6/7 yum安装mysql其它版本(mysql5.7/5.6)

    对于mysql,是越来越不想花时间安装了,但yum的自带版本太低,必须简单处理下。 yum list installed | grep mysql #移除已安装的 yum remove mysql-l...

    2020-09-29 15:41:10 2872
  • Centos一键安装部署vsftp

    Centos一键安装部署vsftp

    每次给不同的人配置ftp,都手工搞一次,决定简化这一步操作。 1、自动安装vsftp 2、自动设置被动模式、并打开iptables端口 3、支持自动创建ftp目录 帐号 密码 4、安装后,以后每次运行...

    2016-10-31 14:22:13 3408
  • windows server 2012 r2激活工具

    windows server 2012 r2激活工具

    最近倒腾下windows server 2012 r2,一直没成功激活,放了大半个月,今天睡觉前再看下,找到了个能用的激活工具,就记录下。 工具名称应该是 KMSpico ,从https://www....

    2020-04-14 15:21:24 11622