在Linux服务器上安装并配置PyTorch环境

前提:首先已经安装好了anaconda。

常用命令:

查看cuda版本:(11.2)

nvidia-smi

查看conda版本:(4.10.3)

conda -V

查看anaconda版本:(1.9.0)

anacanda -V

创建文件夹:

mikdir packeges

查看都有什么:

conda evn list

查看当前目录的所有文件夹:

ls

进入某个包:

conda activata packeges

python、pytorch、torchvision版本对应关系:

https://blog.csdn.net/qq_43599739/article/details/123015331

一、安装pytorch

1、创建虚拟环境

由于自己在不同项目框架中所需要的环境配置基本上都不相同,所以就需要为每个项目创建一个独立的虚拟环境,这样才不会造成互相冲突,也是为了方便管理,这也是要安装Anaconda 的原因。【使用Anaconda 管理多个不同的配置环境】

# pytorch_3.7是环境名、python=3.7是python的版本,这两个可以改变。
conda create -n pytorch_3.7 python=3.7

输入:y

 安装成功:

 2、激活环境

conda activate pytorch_3.7

 【补充】退出环境

conda deactivate

【补充】查看环境

conda info --envs
# 或者
conda evns list

【补充】删除环境

conda evn remove -n pytorch_3.7

3、找到合适的pytorch版本

进入pytorch官网:

https://pytorch.org/

设置清华源,加速安装。

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

官网没有查到cuda11.2对应版本,只查到cudatoolkit=11.0的。

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

参考资料:https://zhuanlan.zhihu.com/p/358723664

但是自动安装上了不需要的版本的numpy之类的:

 查看是否安装成功:

conda list

 再次验证:

python
import torch
torch.cuda.is_available()

 退出python环境:

exit()

4、卸载版本不对的numpy等、重装numpy

卸载numpy:

pip uninstall numpy

安装numpy:

pip install numpy==1.20.3

 

 查看都安装了什么:

conda list

 全文主要参考了https://blog.csdn.net/Thebest_jack/article/details/125493728

物联沃分享整理
物联沃-IOTWORD物联网 » 在Linux服务器上安装并配置PyTorch环境

发表评论