安装pytorch 方法:
conda create -n pytorch mypython =3.7 创建环境变量mypytorch
搜索 https://PyTorch
pytorch 版本选择 如下
conda activate mypytorch
pytorch网站里面的连接:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
查看是否安装成功:
conda activate mypytorch
python
import torch
torch.__version__ #pytorch版本
torch.cuda.is_available() # 查看GPU版本是否安装成功
可以看看视频了解一下: