欢迎光临散文网 会员登陆 & 注册

win11装 Pytorch, RFDiffusion

2023-09-01 22:43 作者:磁通量d  | 我要投稿

事没怎么干软件装一堆。。主要参考:

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 <配置的虚拟环境名>






win11装 Pytorch, RFDiffusion的评论 (共 条)

分享到微博请遵守国家法律