创建虚拟环境:
先创建一个虚拟python环境,敲如下代码
然后再到该虚拟环境里面安装自己想要的包
激活虚拟环境
然后再聚类训练这些
验证GPU加速
阿里源
cpp
pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com