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?
來建虛擬環(huán)境會卡在?solving 上。。所以手動配置虛擬環(huán)境。。
RFDiffusion 依賴于pytorch,which 依賴于 CUDA
裝CUDA:
顯卡驅(qū)動版本,在cmd里輸入nvidia-smi,版本不要高于右上角顯示版本。
sudo?版本太低了,在官網(wǎng)裝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
建立虛擬環(huán)境,除了用主機(jī)的配置外還可以在環(huán)境里分別裝配置;后悔沒早點(diǎn)裝,否則之前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)虛擬環(huán)境
conda?deactivate 退出虛擬環(huán)境
如果顯示找不到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。。
配置
進(jìn)入clone?的RFDiffusion文件夾建立虛擬環(huán)境:
conda?create?--name?rfd?python=3.9
conda?activate?<配置的虛擬環(huán)境名>
裝 PyTorch
nvcc?--version??可以輸出版號就不在虛擬環(huán)境裝cuda;根據(jù)cuda版本選擇pytorch版本
找合適的pytorch版本:https://pytorch.org/get-started/locally/
pip3?install?torch?torchvision?torchaudio?--index-url?https://download.pytorch.org/whl/cu117
根據(jù)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__) 輸出應(yīng)該為?dgl-1.1.2+cu117(+cu為必須),否則運(yùn)行報(bào)錯dgl._ffi.base.DGLError:?Operator?Range?does?not?support?cuda?device.
裝錯先卸載再裝:
pip?uninstall?dglgo?dgl
配置好后
每次使用時進(jìn)入RFDiffusion文件夾,輸入:
conda?activate <配置的虛擬環(huán)境名>