win11装 Pytorch, RFDiffusion
事没怎么干软件装一堆。。主要参考:
https://github.com/RosettaCommons/RFdiffusion#getting-started--installation
git clone https://github.com/RosettaCommons/RFdiffusion.git
然后跟着 readme 一直到 conda
如果直接用 conda env create -f env/SE3nv.yml
来建虚拟环境会卡在 solving 上。。所以手动配置虚拟环境。。
RFDiffusion 依赖于pytorch,which 依赖于 CUDA
装CUDA:
显卡驱动版本,在cmd里输入nvidia-smi,版本不要高于右上角显示版本。
sudo 版本太低了,在官网装11.7
https://developer.nvidia.com/cuda-toolkit-archive
如果用sudo装了又卸了usr/bin 里会没有nvcc命令,更改 .bashrc添加路径:
export PATH="/usr/local/cuda-11.7/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH"
然后要更新一下
source ~/.bashrc
装 CONDA
建立虚拟环境,除了用主机的配置外还可以在环境里分别装配置;后悔没早点装,否则之前ubuntu大概可以少卸载几次。。
conda库: https://repo.anaconda.com/miniconda/
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
conda update conda
source ~/miniconda3/bin/activate 会开启(base)虚拟环境
conda deactivate 退出虚拟环境
如果显示找不到conda命令:
vim ~/.bashrc
添加:
export PATH="${PATH}:<miniconda3/bin路径>"
如果出错要完全卸载再重装,不然安装可能会出错。
更改 .bashrc后要
source ~/.bashrc
更新一下或者重启
卸载:
rm -r <miniconda3文件夹>
cd ~/
ls -a 列出隐藏文件
rm -r .conda
rm .condarc
显示 channel(源):
conda config --set show_channel_urls yes
conda config --show-sources
添加源:
conda config --add channels 。。。
不怎么需要改。。用 pip 装更方便。。顺便清华源会连不上404。。
配置
进入clone 的RFDiffusion文件夹建立虚拟环境:
conda create --name rfd python=3.9
conda activate <配置的虚拟环境名>
装 PyTorch
nvcc --version 可以输出版号就不在虚拟环境装cuda;根据cuda版本选择pytorch版本
找合适的pytorch版本:https://pytorch.org/get-started/locally/
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
根据env/SE3nv.yml手动安装依赖库:
pip install hydra-core pyrsistent torchaudio torchvision omegaconf
找合适的dgl版本 https://www.dgl.ai/pages/start.html
pip install dgl -f https://data.dgl.ai/wheels/cu117/repo.html
检查pytorch, dgl:
python
import torch
print(torch.__version__)
import dgl
print(dgl.__version__) 输出应该为 dgl-1.1.2+cu117(+cu为必须),否则运行报错dgl._ffi.base.DGLError: Operator Range does not support cuda device.
装错先卸载再装:
pip uninstall dglgo dgl
配置好后
每次使用时进入RFDiffusion文件夹,输入:
conda activate <配置的虚拟环境名>