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


多套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



相关推荐

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

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

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

猜你喜欢

2分钟快速安装部署mysql5.7

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

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

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

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

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

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

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