Linux服务器配置pytorch1.9.0

简介

因为要跑代码，需要在学校的服务器上配置python3.6 + pytorch1.9.0，已有Anaconda和cuda(10.2)

步骤

1，创建环境

创建一个名为torch19_py36的虚拟环境：

conda create -n torch19_py36 python=3.6

激活该环境，接下来对应的安装操作都是在该环境上进行

conda activate torch19_py36

2，配置pytorch

因为要安装pytorch1.9.0（目前最新已经到1.12了），在官网可以看到老的版本对应的操作：previous-versions

我们要装的是1.9.0，cuda版本为10.2

conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch

由于是学校服务器没法用代理，下载速度很慢，所以考虑使用离线安装的方式，ctrl + c 即可中断当前包的安装，这里使用离线的方式来安装torch和torchvision这两个比较大的包

这里可以找到pytorch的各个版本torch
点进去是不同版本的torch和torchvision，cpu表示CPU版本的Pytorch，cp表示Python版本，cu是cuda版本

这里要找到我们所需要的torch（python3.6 + cuda10.2），torchvision同理

下载，并上传至服务器

接下来，安装这两个包，torch要在torchvision之前安装

pip install torch-1.9.0+cu102-cp36-cp36m-linux_x86_64.whl

pip install torchvision-0.10.0-cp36-cp36m-linux_x86_64.whl

（这里有可能需要安装或升级numpy，这也没什么，直接安装就行）

3，测试

启动python查看是否安装成功