服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision

接着上一篇安装完anaconda后,就可以一步一步来搭建我们炼丹需要的环境了

因为我目前在做点云配准,这次呢,就以pcrnet为例,我们来搭建环境

pcrnet源码地址: https://github.com/vinits5/pcrnet_pytorch

pcrnet文章地址: https://arxiv.org/abs/1908.07906

pcrnet需要的环境(源码一般会告诉你需要啥)

服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision

那我们就先安装torch和torchvision吧

1.找到对应版本信息 

https://www.cnblogs.com/lyl0618/p/13323010.html

服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision

2. 找到源

阿里云 https://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣(douban) https://pypi.douban.com/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学 https://pypi.mirrors.ustc.edu.cn/simple/

刚开始用的清华源,太慢了,老是报错openssl啥的,

服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision

其实就是网太慢了,换了一下阿里云,下的飞快

输入: pip install  torch==1.6.0  torchvision==0.7.0 -i  https://mirrors.aliyun.com/pypi/simple/

如果报错  NotWritableError: The current user does not have write permissions to a required path.

意思就是没得权限,那就给他权限

$ cd ../    (回退到当前目录的上一级)

$ sudo chmod o+w dirname  (dirname为当前目录的名字)

3. 测试torch安装成功没有

服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision

出现True表示ok

4. 下载其他的库

  1. transforms3d==0.3.1
  2. h5py==2.9.0

如果不能用conda,就用pip,两个都可以下载,(用的阿里源)

5.下载源代码

先cd到我想保存的地方 cd /home

然后 git clone https://github.com/vinits5/pcrnet_pytorch.git

报错fatal: could not create work tree dir ‘xxxx’: Permission denied

解决办法:

https://blog.****.net/one_girl/article/details/82887067

解决办法就是先修改目录的权限然后在进行操作

如果有密码的可能会提醒你输入密码,然后就可以重新执行git clone命令了

6.运行源码

cd /home/pcrnet_pytorch

根据教程,输入 python train_pcrnet.py

会报错 没得 tensorboardX这个module

不要慌 下载(conda下载不了这个库,我用pip,直接用也报错openssl,指定阿里源后,下的飞速)

pip install  tensorboardX -i  https://mirrors.aliyun.com/pypi/simple/

下载完后,重新运行,他会先下载数据,很慢很慢很慢,我先回去了,不管他,明天再来看

服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision