1.由于这个容器拉去下来后自带python3.8,所以我们需要安装对应的pytorch版本,首先我们需要更新一下pip,否则用不了安装命令
apt update
apt install python3-pip
2.安装pytorch(这个torch是我针对python3.8查的版本 ,很好用)这个过程可能有点慢……
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
3.检验是否安装成功(检验是否能用GPU帮助训练)
python
import torch
torch.cuda.is_available()
如果输出True,说明没问题
接下来就可以按照你的复现实验来继续往下进行了~!祝科研顺利^_^