在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

出现这个页面就成功了
image

2.安装cuda,检验cuda

nvcc --version

出现这个页面就成功了
image
提示一下,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

出现这个页面就成功了
image

4.安装Anaconda,检验

conda -V

出现这个页面就成功了
image

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()
pip install scikit-learn -i https://pypi.tuna.tsinghua.edu.cn/simple/

只要出现true就行了,如果是false建议重启一下就可以了。或者conda list看一下版本

7.去pycharm配置环境就行了

结束

注意

保持心态,这一周不知道了试了几版,先在GPU机子上装了win10然后去试了装个虚拟机,但是因为磁盘分配问题没有考虑到位,也引起了对后续显卡调用问题的思考,最终决定放弃win10装虚拟机的想法,敲定了GPU装原生态的ubuntu,找一台机子重新装了一台win10.安装过程总会遇到各种各样各样奇怪的事情,这些是我能大部分想起来的事情了,已经能解决掉很多问题了,这期间同门和师兄师姐都在做好多事情,我也要努力追赶上。

本文作者:太好了还有脑子可以用

本文链接:https://www.cnblogs.com/ZarkY/p/17706027.html

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   太好了还有脑子可以用  阅读(167)  评论(1编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起