一、前提准备
Win10,已下载安装Python以及Anaconda、PyCharm,有独显。
查看cuda版本:
- 搜索框搜索NVIDA控制面板,打开该界面
- 依次点击帮助→系统信息→组件
二、步骤
1.PyTorch版本选择
进入PyTorch官网:https://pytorch.org/ ,点击GET STARTED,选择电脑配置的对应版本,查看安装PyTorch的命令
pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
打开Anaconda Powershell Prompt,粘贴命令,下载安装即可。
2.检查是否安装成功
在命令行输入pip show torch查看是否安装成功,以及安装信息。
同时,PyCharm也可以查看已安装的包