ubuntu利用conda创建虚拟环境,并安装cuda,cudnn,tensorflow

1.安装NVIDIA驱动:

CUDA版本与显卡驱动版本匹配查询:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html/
ubuntu利用conda创建虚拟环境,并安装cuda,cudnn,tensorflow_第1张图片
NVIDIA驱动下载地址:https://www.geforce.cn/drivers
根据自己的显卡型号以及操作系统下载对应驱动

2.conda安装仓库

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

3.创建虚拟环境

conda create -n your_env_name python=3.6,若不写依赖项python的版本号,则默认安装最新的python包

4.激活虚拟环境

conda activate your_env_name,只有激活虚拟环境,你才能使用虚拟环境

5.安装cuda和cudnn

tensorflow、cuda和cudnn对照表如下:
ubuntu利用conda创建虚拟环境,并安装cuda,cudnn,tensorflow_第2张图片

5.1 安装cuda

conda install cudatoolkit=10.1

5.2 安装cudnn

conda install cudnn=7.6.5

6.安装tensorflow-gpu

pip在线使用阿里源安装
pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow-gpu==2.1

pip永久使用阿里源:
cd $HOME
vi .pip/pip.conf

[global]
index-url = https://mirrors.aliyun.com/pypi/simple

你可能感兴趣的:(conda)