在服务器上配置pytorch环境,Cuda、cudnn什么的都是别人安装好的。所以我只是装个Pytorch,然而也是失败了好几天,加上服务器是内网,没法直接ssh访问,东西scp没法传上去,web和ip也不能访问T-T。折腾了两天···
服务器环境:Ubuntu18.04,Cuda10.02,cudnn5.7.0?
首先创建conda环境,我是以前创建好的,具体创建方法可以百度~
按照官方安装出现问题:
1、使用pip安装Pytorch,上官网~
Previous PyTorch Versions | PyTorch https://pytorch.org/get-started/previous-versions/2、找到对应的版本,可以选择conda安装也可以pip,我这里要安装cuda10.02版本,torch1.9.0的。
一直提示网络错误或者没有找到所需要的包~尝试换了镜像源,嘿~失败万岁~
如果成功了就不用往下看了~
打开下面这个地址看了下,其实pip安装就是下载whl文件然后安装~手动找一下,可以看到有:
官网下载库地址:
https://download.pytorch.org/whl/torch_stable.html
https://download.pytorch.org/whl/cpu/torch-1.9.0%2Bcpu-cp36-cp36m-linux_x86_64.whl
cu102是指cuda10.02,cp36可能是python3.6?linux系统,64位。
这不就好办了,wget下下来安装~BUT···服务器访问它还是会报错,没办法,只好下载到本地,再上传到租来的web服务器,让显卡服务器从web服务器下载~
总之,就可以把包下载到本地啦~
wget https://download.pytorch.org/whl/cu102/torch-1.9.0%2Bcu102-cp36-cp36m-linux_x86_64.whl
然后同样的找到官方给的代码里面的版本,除了刚才的torch 1.9.0,还有torchvision 0.10.0(注意也要找到是‘cu102/xxxxx.whl’的),还有torchaudio 0.9.0(注意这个不用cu102了,当然有的可以试试,没试过)
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
所以可以找到如下:
#下载 torch 1.9.0#
wget https://download.pytorch.org/whl/cu102/torch-1.9.0%2Bcu102-cp36-cp36m-linux_x86_64.whl
#下载 torchvision 0.10.0#
wget https://download.pytorch.org/whl/cu102/torchvision-0.10.0-cp36-cp36m-linux_x86_64.whl
#下载 torchaudio 0.9.0#
wget https://download.pytorch.org/whl/torchaudio-0.9.0-cp36-cp36m-linux_x86_64.whl
下载完成~
注意,如果需要把它安装到某个conda环境下,要先进入conda环境,我现在在base,因此需要 conda activate 环境名字 来激活环境,接着才进行安装!
重要提示:接下来的安装顺序必须首先安装torch(不知道为啥,别人说的)
重要提示:接下来的安装顺序必须首先安装torch(不知道为啥,别人说的)
重要提示:接下来的安装顺序必须首先安装torch(不知道为啥,别人说的)
开始安装
#激活环境,我的是p36t15,要替换为你自己的#
conda activate xxxxx
#cd到whl安装包所在目录下
cd /xxx/somepackage/
#依次安装torch,torchvision,torchaudio#
pip install torch-1.9.0+cu102-cp36-cp36m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torchvision-0.10.0-cp36-cp36m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torchaudio-0.9.0-cp36-cp36m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
安装完成,测试
$ python
Python 3.6.13 |Anaconda, Inc.| (default, Jun 4 2021, 14:25:59)
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>import torch
>>>torch.cuda.is_available()
True