ubuntu20.04下不同版本的pytorch切换使用不同版本cuda

• ubuntu20.04 (pytorch1.6+cuda10.2) & (pytorch1.2+cuda9.2)   多cuda切换
• neural_renderer_pytorch在torch1.2下运行,1.6装不上
• 使用命令切换cuda版本

  sudo rm -rf cuda
  sudo ln -s /usr/local/cuda-9.2 /usr/local/cuda
  
  sudo rm -rf cuda
  sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda

最近想学习 无先验三维模型,无监督,利用对称信息的三维重建代码 2020 cvpr https://github.com/elliottwu/unsup3d

使用微分渲染器, 我当前的环境为 pytorch1.6, cuda 10.2, 用如下命令安装, 源码为https://github.com/daniilidis-group/neural_renderer

pip install neural_renderer_pytorch
或者 pip install https://github.com/daniilidis-group/neural_renderer/zipball/master
两条命令一样的

 报错很多,如下

搜索报错原因, 需要将pytorch版本降低到1.2.0(见https://zhuanlan.zhihu.com/p/195068015) 同时unsup3d的作者写到环境为pytorch1.2.0&cuda9.2, 而我的版本是pytorch1.6, cuda 10.2

那就重新创建一个虚拟环境安装torch1.2.0(https://pytorch.org/get-started/previous-versions/) 官方推荐的对应cuda版本为9.2和10.0, 而我的是10.2, 搞个10.2