在Ubuntu20.0下搭建CUDA、cuDNN、Anaconda、pycharm

其他链接

1.buntu18.04下搭建CUDA、cuDNN、Anaconda、tensorflow1.15、Pycharm、ros、Cl
2.pip使用中科大源、清华源或修改默认源为中科大源、清华源

安装

照着链接1的方法就可以下好cuda、cudnn和anaconda

1.先装驱动，检验驱动

nvidia-smi

出现这个页面就成功了

2.安装cuda，检验cuda

nvcc --version

出现这个页面就成功了

提示一下，duda和Anaconda都需要配置文件

①打开配置文件

sudo gedit ~/.bashrc

②最后一行输入

自己改自己的路径

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/home/grchww/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/home/grchww/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/home/grchww/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/home/grchww/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
 
# <<< conda initialize <<<
 
#Anaconda
export PATH="/home/grchww/anaconda3/bin:$PATH"
 
#CUDA
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

3.安装cudnn，检验

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出现这个页面就成功了

4.安装Anaconda，检验

conda -V 

出现这个页面就成功了

5.安装pycharm

根据这个链接就可以下载

6.构建虚拟环境，下载pytorch

可以看我这篇博客，linux跟win10配置pytorch一样
注意，下载的torch版本不一致，因为此时这里是gpu版本，如果照常下载可能会下cpu版本的pytorch，所以这里需要更换源，然后在下载就行了

①更换源

选择合适的版本
pip使用中科大源、清华源或修改默认源为中科大源、清华源

②下载

我的版本是cuda10.2和cudnn7.65

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html