服务器配准深度学习环境(二)—— 搭建torch和对应版本的torchvision
接着上一篇安装完anaconda后,就可以一步一步来搭建我们炼丹需要的环境了
因为我目前在做点云配准,这次呢,就以pcrnet为例,我们来搭建环境
pcrnet源码地址: https://github.com/vinits5/pcrnet_pytorch
pcrnet文章地址: https://arxiv.org/abs/1908.07906
pcrnet需要的环境(源码一般会告诉你需要啥)
那我们就先安装torch和torchvision吧
1.找到对应版本信息
https://www.cnblogs.com/lyl0618/p/13323010.html
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啥的,
其实就是网太慢了,换了一下阿里云,下的飞快
输入: 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安装成功没有
出现True表示ok
4. 下载其他的库
- transforms3d==0.3.1
- 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/
下载完后,重新运行,他会先下载数据,很慢很慢很慢,我先回去了,不管他,明天再来看