ubuntu安装cuda,cudnn,pytorch

1.conda创建虚拟环境

conda create -n pytorch pip python=3.8

激活虚拟环境

source activate pytorch

2.虚拟环境中安装cuda和cudnn

添加国内镜像

添加清华源
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
并设置搜索时显示通道地址：
conda config --set show_channel_urls yes

注：使用以下命令恢复默认源
conda config --remove-key channels

安装cuda

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

也可用：

conda install cudatoolkit=10.1
也可以不带通道，因为前面已经配置了通道，使用conda命令安装是，会自动从配置的通道中找安装源