bert pytorch源码_RTX 3090的深度学习环境配置pytorch、tensorflow、keras

笔者中山大学研究生，医学生+计科学生的集合体，机器学习爱好者。

最近刚入了3090，发现网上写的各种环境配置相当混乱而且速度很慢。所以自己测了下速度最快的3090配置环境，欢迎补充！

基本环境（整个流程大约需要5分钟甚至更少）

py37或py38
cuda11.0
cudnn8.0.4
tf2.5（tf-nightly）
pytorch1.7
keras2.3

（1）官网下载，安装显卡驱动：

bash NVIDIA-Linux-x86_64-455.23.04.run

（2）安装Anaconda并换源

bash Anaconda3-5.2.0-Linux-x86_64.sh
vim ~/.bashrc
export PATH=/home/XXX/anaconda3/bin:$PATH（XXX为自己的用户名）（在文件末尾处添加该语句）
source ~/.bashrc

conda config --add channels https://mirrors.t