暂时关闭所有代理
unset http_proxy && unset https_proxy
查看实验室卡的占用情况
watch -n 1 nvidia-smi
需要cuda版本和实验室版本匹配,因此要装对应cuda版本的pytorch
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
查看的方法如下
conda activate xxx
python
import torch
torch.__version__
conda list