• 在dataloader里面加上multiprocessing_context=”forkserver”
  • 在主函数运行
name: cs224n-gpu
channels:
- defaults
- nvidia
dependencies:
- python==3.10
- nvidia::cuda-toolkit==12.1.1
- pip
- pip:
- -r requirements.txt

下载的时候清华源找不到nvidia, 在condarc里面加上

channels: - defaults 
show_channel_urls: true 
channel_alias:https://mirrors.tuna.tsinghua.edu.cn/anaconda 
default_channels: 
-https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main 
-https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r 
custom_channels: 
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud 
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud 
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
nvidia: https://conda.anaconda.org/ (add)