admin管理员组

文章数量:1794759

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

服务器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版本对应关系:

blog.csdn/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官网:

pytorch/

设置清华源,加速安装。

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

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

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

参考资料:zhuanlan.zhihu/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

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

本文标签: 系统安装环境服务器Linuxpytorch