简介
因为要跑代码,需要在学校的服务器上配置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查看是否安装成功