问题描述
在vscode中想用命令行执行代码:
python train.py --gpu_ids 0 --name ft_ResNet50 --train_all --batchsize 32 --data_dir your_data_path
结果显示如下:
vscode运行环境是对的,这个环境中是有torch的,但是命令行显示当前环境中没有torch。
原因分析:
尝试1
我尝试直接在vscode命令行输入
python
结果如下。这个python3.8是base环境下的(我的pytorch环境python版本是3.9)
尝试2
想到应该是我的环境不同,所以尝试在命令行激活环境。
conda activate pytorch_gpu
再次尝试输入
python
失败。结果还是base环境中python版本。
解决方案:
问题出在终端是shell还是cmd
在这个地方选择cmd,命令行出现如下形式。
这样就可以了,前面出现了环境的名称。