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

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

2024-01-10 10:55:48安装部署 阅读 186


多套AI机器学习框架最佳的实践方法就是使用conda来管理不同的虚拟环境,以前用pip喜欢手动下载手动安装,什么cuda cudnn,各种版本相互依赖,conda就专治这个。但由于习惯,在使用conda之后还习惯性地在conda中做pip安装cuda cudnn,直到后来完全了解了conda install能够根据显卡型号和python版本和要安装的ai框架版本自动解决依赖问题。

1、安装conda,这个似乎没有太多强制版本的,可直接从官网下载最新。

conda下载地址

https://docs.conda.io/en/latest/miniconda.html有时用迅雷等下载工具比浏览器快一些

安装完成后以管理员方式运行 conda3



2、配置源,这主要是配置国内源达到下载加速

以管理员方式运行 conda3后 在conda的窗口上运行命令

conda config --set show_channel_urls yes

之后会在 C:\Users\(电脑用户名)会生成一个配置文件.condarc 打开编辑


channels:
 - defaults
show_channel_urls: true
default_channels:
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
 conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
编辑保存后,运行conda clean –i

以下是两种机器学习框架的安装过程,tf已经实践过,pytroch 尚未实践,根据个人了解补充。

安装tensorflow-gpu==1.15 

退出之前的环境,conda deactivate

conda create -n tf15 python=3.7 #创建环境
activate tf15 #进入环境
conda install tensorflow-gpu=1.15.0#安装tf gpu版 会自动安装对应的cuda和cudnn
这里tf1.x的python版本不能高于3.7,如果创建了高于3.7的python安装会报错。

如果不指定版本,就不论python版本,全按最新。

现在tf已经出到2.2x了,少量代码会强制要求tf版本。

同时tf1.15之后,不再区分gpu和cpu,不再存在tensorflow-gpu这种名称了。


安装pytroch 

退出之前的环境,conda deactivate

官网查看

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


conda create -n pytorchenv python=3.8.10
activate pytorchenv
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia



相关推荐

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

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


本文《conda部署与物理机不同的cuda版本,一行命令部署pytroch或TensorFlow环境》由爱思考吧 isres.com 分享,转载请注明出处。

本文网址:https://www.isres.com/bushu/211.html

本站主要收集测评能够节省时间和提升效率的软件工具并分享使用与学习的过程和感受,任何情况下用户都需遵守所使用软件资源的相关协议。

与《conda部署与物理机不同的cuda版本,一行命令部署pytroch或TensorFlow环境》有关的本站文章及资源仅供个人学习交流使用,相关资源请下载后24小时内删除,请勿用于其它用途,因此产生的任何问题由您自行承担。

猜你喜欢

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

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

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

    2023-09-07 09:42:20 790
  • Linux安装php imagick扩展

    yum install autoconf gcc gcc-c++ ImageMagick-devel wget https://pecl.php.net/get/imagick-3.4.4.tgz ...

    2023-03-23 16:18:16 437
  • 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 717
  • pytroch AI机器学习环境部署,RTX3080显卡

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

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

    2023-04-24 12:37:24 470
  • 给PhpStorm简单部署一个运行php代码的环境

    给PhpStorm简单部署一个运行php代码的环境

    1、下载Php https://windows.php.net/download#php-7.4-nts-vc15-x86 本次demo中下载的是 https://windows...

    2023-06-07 09:37:44 321
  • LuaJIT-2.1.0相关以及ngx_lua_waf防火墙配置相关

    wget https://www.isres.com/file/LuaJIT-2.1.0-beta3.tar.gz tar -zxvf LuaJIT-2.1.0-beta3.tar.gz cd Lu...

    2018-01-20 14:44:41 3089
  • 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 423
  • 2分钟快速安装部署mysql5.7

    今天分享mysql5.7在Linux的安装部署,服务器运维工作都会涉及到安装部署mysql5.7。 mysql5.7在Linux的安装一般有三个方法,分别是: 1、源码编译安装 2、rpm包安装 3、...

    2023-04-24 11:53:13 437