Linux服务器上的pytorch配置


之前在服务器上配置过一次,但由于被师兄误删了,最近又要用到,就去重新配置了一下。上回配置,碰到好多问题,把自己都搞乱了,所以也没记录下什么,就趁着这次机会详细记录一下。

Anaconda安装

安装anaconda是为了创建虚拟环境便于管理。去官网或者清华镜像站下载对应版本的anaconda。
因为服务器的网速有点慢,所以选择了本地下载好上传到服务器的方式。anaconda的本地文件在桌面,因此直接右击,选择Git Bash here,再输入以下命令:

scp -r Anaconda3-2020.07-Linux-x86_64.sh(即文件名) username@serverip:/home/username/目的文件夹名

安装命令:

bash Anaconda3-2020.07-Linux-x86_64.sh

验证是否安装成功:重新打开命令界面,输入命令conda --version,安装成功就会显示conda的版本信息。命令python3quit可查看anaconda自带python的版本信息。conda update conda更新conda。conda --help或者conda -h查看conda帮助。

创建虚拟环境:

conda create --name 环境名(建议英文) 包名

例如conda create --name python37 python=3.7,即创建一个名为python37的环境,环境中安装版本为3.7的python。--name可以替换为-n

切换conda环境的命令:conda activate 环境名
退出虚拟环境:conda deactivate
显示安装过的所有虚拟环境:conda info --envs或者conda info -e或者conda envs list
复制环境:conda create --name 新环境名 --clone 被复制的环境名
删除环境:conda remove --name 环境名 --all
获取当前环境中已安装的包信息:conda list
指定环境安装包:conda install --name 环境名 要安装的包名
在当前环境中安装包:conda install 要安装的包名

如果conda安装不上的包,或者没有的包,可以使用pip安装:pip install 安装的包名
安装完anaconda之后,会发现Linux的终端变样了,在终端前部分出现了(base)字样。这是因为,在我们打开终端的时候,自动执行了conda activate base命令,而执行conda deactivate命令即可去掉(base)字样。

查看服务器的cuda版本

使用下面这条命令检查电脑上有哪些显卡,安装了什么驱动。

lspci -k | grep -A 2 -i "VGA"

然后查看什么驱动是系统推荐安装的,我们需要的是nvidia产家的驱动。

sudo ubuntu-drivers devices

最好安装后面显示recommended字样的显卡驱动。

查看服务器cuda版本的命令

ncvv -V

由于服务器暂时还没有NVIDIA独立显卡,因此省略NVIDIA显卡驱动安装以及cuda和cuDNN的安装。

pytorch安装

进入pytorch官网,选择合适的版本安装。
在这里插入图片描述
输入合适版本显示的命令前,先切换到实现创建的虚拟环境,再输入安装命令。这样就会把pytorch安装到你创建的环境上。

安装过程比较慢,需要耐心等待!!!若是安装过慢,可以使用清华源进行安装。
检查是否安装成功:

>>> import torch
>>> torch.__version__

显示pytorch的版本即为安装成功。

通过以下命令测试是否可以正常导入库和包。

>>>import torch
>>>import torchvision
>>>import cv2
>>>import numpy
>>>import matplotlib
>>>import dominate
>>>import random
>>>import collections
>>>import shutil

如果没有报错,则环境配置成功

卸载pytorch

conda uninstall pytorch

其他

  • pip安装和conda安装的区别
    pip:不一定会展示所需其他依赖包,在安装包时有时会直接忽略依赖项而安装,仅在结果中提示错误。
    conda:列出了所需其他依赖包,在安装包时会自动安装其依赖项,另外可以便捷地在包的不同版本中自由切换。

  • 服务器换源
    例如换成清华源,命令如下:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

输入这两行命令即可换源成功。但要注意,换源成功后,一旦更新库就会把所有的库再更新一遍。换源成功后,输入以下命令进行更新:

conda update –all(更新所有库)
conda update conda(更新 conda 自身)

opencv的安装命令

conda install -c menpo opencv
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值