栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Ubuntu下安装Anaconda3、tensorflow-gpu

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu下安装Anaconda3、tensorflow-gpu

Ubuntu下安装Anaconda3、tensorflow-gpu(RTX3080Ti) 一、安装Anaconda#

从官网下载[Linux版本的Anaconda3],选择自己想要下载的版本即可(https://www.anaconda.com/products/distribution#Downloads)

下载完成后通过指令“bash 文件路径/Anaconda3-2021.11-Linux-x86_64.sh”就可以开始安装了,安装过程中,需要输入两个“yes”,第一个是是否接受许可条款,输入yes

然后会询问是否要修改安装路径,不修改的话,直接enter就开始安装

安装好需要的一些包之后,会询问是否要初始化环境,输入yes,就不需要自己再去加环境变量了

如果不小心输成no了,也可以自己vim ~/.bashrc 在最后一行加入环境变量即可
export PATH=$PATH:【你的安装目录】
最后保存更改:source ~/.bashrc
出现下面这个界面就说明安装成功了

切换到Anaconda3安装目录下的bin,用source activate base激活conda,不然直接使用conda命令,会提示“未找到命令”。这一步之后就可以开始你的任何操作了!

二、创建虚拟环境,并安装torch
  1. 创建虚拟环境
    conda create -n envname python=3.X
    创建好之后,也会给出激活和退出的命令

    使用 conda activate envname进入刚刚创建的虚拟环境,安装的tensorflow、torch等包就安装到了自己的环境中了。

  2. 安装torch
    进入pytorch官网所有版本的下载地址,选择一个符合自己GPU算力的版本,复制命令到命令行中即可

    使用上面的命令安装,可以直接把pytorch和cuda都装好,所以如果pytorch和tensorflow都经常用的朋友,现状pytorch可以更省事哦。

  3. CUDA算力查看
    在本机中安装了cuda,则切换到如下目录中去查看对应的cuda驱动版本和算力大小,RTX3080Ti支持的驱动版本是11.3,对应算力为8.6,所以在安装CUDA时需要安装支持算力为8-8.6的版本。10.1是当前运行的CUDA版本,之前安装到本机上的。
    也可直接用命令“nvidia-smi”查看

    RTX3080Ti以上系列的都要装CUDA11.0以上,如果装10.X,在代码运行过程中就不能使用GPU。

  4. 测试是否安装成功
    进入python环境,输入一下命令

    import torch
    torch.zeros(1).cuda()
    

如果输出相应的tensor张量,及设备是CUDA就成功了

如果报错是GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.那么一定是cuda版本的问题了。安装cuda版本不对。

三、安装tensorflow-gpu
  1. 配置cuda和cudnn环境
    安装tensorflow-gpu之前要安装对应的cuda和cudnn,在安装torch包的时候,已经安装好了cuda,在这里只需要选择与上面cuda相应的cudnn就可以了。
    用命令“conda search cudnn -c conda-forge”可查看CUDA支持的cudnn版本,用conda命令安装一下即可

    我们安装与11.3对应的cudnn8.2.1

  2. 安装tensorflow-gpu
    cuda和cudnn都已经安装好了,直接命令行中安装gpu版的tensorflow即可。
    安装命令: “conda install tensorflow-gpu2.6.0”,如果出错,用pip安装即可
    “pip install tensorflow-gpu
    2.6.0”
    下面的环境在在GPU为RTX3080Ti中是亲测可用的

    	cuda ==11.0或11.1
    	cudnn ==8.2.1
    	python ==3.6/3.7#没有用到3.8和3.9更高的版本,应该也是没问题的
    	TensorFlow-gpu ==2.5.0
    	Pytorch ==1.7.1#对应cuda11.0
    

在RTX3090Ti中,各版本稍微不一样,需要注意的一点是,tensorflow-gpu要安装2.X,安装1.X一样会出错。

		cuda ==11.3
		cudnn ==8.2.1
		python ==3.9#没有用到3.8和3.9更高的版本,应该也是没问题的
		TensorFlow-gpu ==2.6.0
		Pytorch ==1.10.1#对应cuda11.3
	```
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/840681.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号