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.011.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
	```
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值