创建python虚拟环境并成功跑通一个模型

在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是默认值。

训练完毕！