服务器Linux系统安装pytorch配置环境

Posted 2023-04-01 Spanish_pie

前提：首先已经安装好了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