在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() 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 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步