安装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’
安装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’