在anaconda3中创建虚拟环境
1.打开终端或者Anaconda Prompt输入
# 创建虚拟环境
conda create -n myenv2 python=3.7
#激活虚拟环境
conda activate myenv2
2.使用pip直接安装pytorch
我电脑的安装的cuda版本是11.5
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu115
这将直接从 PyTorch 官方的 CUDA 11.5 兼容版本的轮子文件中安装 PyTorch。
3.验证安装
安装完成后,你可以通过以下命令验证 CUDA 是否可用:
python -c "import torch; print(torch.cuda.is_available())"
如果输出是 True,说明 CUDA 和 PyTorch 配置正确。
运行项目
1.先给它配置虚拟环境。
2.编辑运行配置
然后编辑运行配置如下,需要导入模型的文件,也需要导入训练的文件。
然后改一下适合你数据的一些地方,例如修改文件路径,分类的个数等等。
还有一个地方需要注意,我在运行过程中,报了这样一个错误。
3.可能报错
RuntimeError: DataLoader worker (pid(s) 11528, 20856, 15176) exited unexpect。
查了一下是因为num_workers是用来指定开多进程的数量,默认值为0,表示不启用多进程。
将num_workers改为0即可,0是默认值。
训练完毕!