自己老是忘记,记一下方便查找(仅针对自己所用的设备资源,部分代码可能不具普适性)。
1)load conda:
spack load minconda3@4.10.3
如果使用module模块的话,则是
module load miniconda3
2)创建并激活虚拟环境,这里python版本可以自由选择,我选的是3.9版本:
conda create -n name python=3.9
conda activate name
3)安装Pytrch,注意安装顺序是torch,torchvision,torchaudio,我使用的cuda版本为11.8,所以此处pytorch安装以版本2.0.0为例,其他对应版本可参考官网:Pytorch
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
4)安装其他包:
pip install -r requirements.txt
附上requirements.txt打包方式,在进入指定环境后,使用freeze进行打包:
pip freeze > requirements.txt