1. 镜像源
一开始把清华镜像源放在的最前面,然后下载各种torch以及想要用的包都容易有依赖冲突。所以把镜像源给改成了 -defaults在前面
vim ~/.condarc
channels:
- defaults
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- conda-forge
show_channel_urls: true
2. 先安装pytorch3d(关键)
先安装pytorch3d(这一步很关键,就是创建好虚拟环境以后直接就这一步,不要先去安装torch等其他东西),而且要指定torch为cuda版本,不然他会默认下载为cpu版本。最好先search一下,指定build来安装。
# 这里得使用pytorch3d的通道,不然找不到可下载依赖
conda search pytorch3d=0.6.0 -c pytorch3d --info
根据自己的python版本, 想安装的cuda版本来选择build
# 根据自己的环境确定build号
conda install pytorch3d=0.6.0=build号 -c pytorch3d
在确认安装以前最好看一下它要下载的torch的build是啥版本的,这样子安装的pytorch3d就是可以用的,不会报Not compile with GPU的错误
3. 其他依赖包不匹配的话
有可能有的依赖包不匹配,比如torchvision会下成cpu版本的,就卸载重新下。
下好以后,运行要跑的程序,缺啥补啥。尽量conda装,conda没有再用pip装。