在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
③检验
python
import torch
torch.cuda.is_available()
只要出现true就行了,如果是false建议重启一下就可以了。或者conda list看一下版本
7.去pycharm配置环境就行了
结束
注意
保持心态,这一周不知道了试了几版,先在GPU机子上装了win10然后去试了装个虚拟机,但是因为磁盘分配问题没有考虑到位,也引起了对后续显卡调用问题的思考,最终决定放弃win10装虚拟机的想法,敲定了GPU装原生态的ubuntu,找一台机子重新装了一台win10.安装过程总会遇到各种各样各样奇怪的事情,这些是我能大部分想起来的事情了,已经能解决掉很多问题了,这期间同门和师兄师姐都在做好多事情,我也要努力追赶上。