创建虚拟环境:
先创建一个虚拟python环境,敲如下代码

然后再到该虚拟环境里面安装自己想要的包

激活虚拟环境
然后再聚类训练这些

验证GPU加速

阿里源
cpp
pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com