安装pytorch的方法是：
conda create -n torch_nightly_env python=3.8
conda activate torch_nightly_env
conda install PyTorch torchvision -c pytorch-nightly
需要在tensor里加入参数 device=‘mps’
m1 pytorch环境安装最靠谱的一种
猜你喜欢
转载自blog.csdn.net/xukeke12138/article/details/125921615
今日推荐
周排行