栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【pytorch】远程服务器搭建深度学习环境miniconda3+cuda11.1+cudnn+pytorch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【pytorch】远程服务器搭建深度学习环境miniconda3+cuda11.1+cudnn+pytorch

好记性不如烂笔头,一定要写文档。。CUDA之前云计算课程装过了,现在也忘记了,pytorch在自己电脑上也安装过,也全忘了。

1. 安装miniconda

conda标准版安装可以查看本文:Linux虚拟机conda环境配置与基本操作

新建了一个conda目录,在该目录下完成操作。

下载命令:

wget -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

使用了清华镜像

安装命令:

bash Miniconda3-latest-Linux-x86_64.sh


不断回车,yes or no输入yes,直至安装完毕。

验证是否安装成功,查看conda版本

conda -V


显示版本即安装成功了


报错:出现命令不存在,不要着急。
参考此教程:https://blog.csdn.net/weixin_38705903/article/details/86533863
输入命令

vim ~/.bashrc

在最后一行加上

export PATH=$PATH:/home/vincent/anaconda3/bin
export PATH=$PATH:【你的安装目录】 

保存修改

source ~/.bashrc

2. 安装并行计算架构CUDA

安装gpu版本之前必须安装并行计算框架CUDA和深度神经网络加速库cuDNN

使用命令查看服务器gpu使用情况,并确定对应的CUDA版本。

nvidia-smi

该命令有效也说明图形显卡驱动已安装,如果显示命令不存在还需要去nvidia官网安装驱动。

根据服务器驱动安装对应的CUDA版本:CUDA安装地址

查看本机Ubuntu系统版本:

lsb_release -a

根据服务器实际情况选择,生成下载命令:

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

选择 continue >> accept >> 空格(取消Driver) >> install

取消驱动安装,Driver按空格取消!!!!



安装完毕 ,配置环境变量。

vim ~/.bashrc


文件最后添加这些内容

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

最后,使之生效

source ~/.bashrc

检测安装结果

nvcc -V

验证cuda是否正常

cd ~/NVIDIA_CUDA-11.4_Samples/1_Utilities/bandwidthTest/
make
./bandwidthTest 

这里最后Result=pass表示cuda显卡带宽测试通过

至此:cuda安装,环境配置和验证完成。


3. 安装cudnn

下载下述三个文件:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111

需要NVIDIA账号,自行注册。

安装命令

sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.1_amd64.deb

查看cudnn版本

dpkg -l | grep cudnn


显示如上即安装成功,安装文件在cuda目录中。


4. 搭建pytorch虚拟环境

创建虚拟环境

conda create -n pytorch-gpu python=3.9


查看当前的虚拟环境

conda info --envs

安装CUDA对应版本的pytorch

下载地址:https://pytorch.org/get-started/previous-versions/

下载命令:

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

吸取上次在本地安装环境的教训,使用了pip安装。


检验是否安装成功

import torch
import torchvision
print(torch.cuda.is_available())


显示True即安装pytorch成功


查看命令汇总 查看CPU信息

command

lscpu

查看GPU信息

Linux查看Nvidia显卡信息及使用情况
Nvidia自带一个命令行工具可以查看显存的使用情况:
command

nvidia-smi

查看操作系统版本
lsb_release -a

参考资料
  1. conda安装:https://blog.csdn.net/wangcassy/article/details/123578709
  2. cuda安装:https://blog.csdn.net/JulyLi2019/article/details/125102098
    https://blog.csdn.net/NCU_Wang/article/details/123612184?utm_source=app&app_version=5.5.0&code=app_1562916241&uLinkId=usr1mkqgl919blen
  3. pytorch-gpu环境安装:https://blog.csdn.net/weixin_44398263/article/details/109110773
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/979726.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号