深度学习服务器环境配置总结

环境配置总结

一、连接服务器

1.1 命令行

登录:

ssh 用户名@服务器IP地址 -p 端口号 然后输入密码就可以了。

缺点是上传下载文件比较麻烦。

1.2 FinalShell

配置好信息即可,上传下载文件方便,有可视化文件管理。

深度学习服务器环境配置总结_第1张图片

二、Pytorch环境配置

2.1 conda

Conda 是一个开源的软件包管理系统和环境管理系统,用于安装多个版本的软件包及其依赖关系,并在它们之间轻松切换。

一般服务器上都先装好了conda,直接创建一个个人虚拟环境。

conda create -n 个人环境名称 python=3.7 ,这样就创建好了一个pyhon版本为3.7的虚拟环境。

激活这个环境:conda activate your_env_name ,就能看到环境由 base 切换到了 your_env_name 。

退出环境:conda deactivate

Conda常用命令:

查看已有环境
conda env list

查看已安装的包
conda list

更改镜像源
conda config --add channels 源地址
conda config --set show_channel_urls yes

删除环境
conda remove -n your_env_name --all

2.2 pytorch 安装

1、在线安装

首先查看GPU的CUDA版本号:nvidia-smi (中间没有空格!)

在 pytorch官网 选择对应CUDA版本、操作系统等,最下面就会给出命令。

深度学习服务器环境配置总结_第2张图片

直接在你的个人环境下运行这条命令即可,遇到网络的问题可以考虑换源。

这里有其他版本:https://pytorch.org/get-started/previous-versions/

2、离线安装

网址:https://download.pytorch.org/whl/cu101

这里是对应cuda10.1版本。找到相应的torch和torchvison文件下载。

torch-1.5.1+cu101-cp37-cp37m-linux_x86_64.whl

torchvision-0.6.1+cu101-cp37-cp37m-linux_x86_64.whl

就比如这两个,cp37代表3.7版本的python。
然后连接服务器,上传这两个文件。激活你的个人环境,在whl文件所在路径下,用命令:pip install 文件名 安装。
注意:torch和torchvison的版本也有对应关系。

对应关系

3、测试

首先进入到你的环境,输入python启动python解释器,再依次输入

import torch
torch.cuda.is_available()

出现True就大功告成,表示能够利用GPU。

你可能感兴趣的:(深度学习,神经网络,机器学习)