Modulus User Guide 安装教程

1.创建环境python3.8

2.安装pytorch，cuda

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

3.安装对应的包

pip install matplotlib transforms3d future typing numpy quadpy\
             numpy-stl==2.16.3 h5py sympy==1.5.1 termcolor psutil\
             symengine==0.6.1 numba Cython chaospy torch_optimizer\
             vtk chaospy termcolor omegaconf hydra-core==1.1.1 einops\
             timm tensorboard pandas orthopy ndim functorch pint -i https://pypi.douban.com/simple/

4.安装torchviz

pip install torchviz -i https://pypi.douban.com/simple/

5.安装GraphViz

conda install GraphViz --channel conda-forge -y

6.安装modulus包，下载modulus包，

git clone git@gitlab.com:nvidia/modulus/modulus.git

需要注册，授权，没有就直接下面链接下载

https://download.csdn.net/download/hungita/87626431

安装：

cd ./Modulus/
python setup.py install

7.测试

Jupyter Notebook workflow - NVIDIA Docs