安装pytorch的方法是:

conda create -n myenv python=3.8
conda activate myenv
conda install PyTorch torchvision -c pytorch-nightly

需要在tensor里加入参数 device=‘mps’

安装pytorch:

conda install pytorch torchvision torchaudio -c pytorch

删除虚拟环境

conda remove -n your_env_name --all
Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐