如果已经安装了pytorch版本,对应跳过步骤一,直接安装步骤二即可
注意:依赖包的配置需要根据pytorch版本号以及cuda的版本号,版本号要适配,否则部分函数无法使用
基于代码安装适配的pytorch版本
通过官网进行安装:pytorch下载官网,此处版本根据代码需要,具体格式如下
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
京公网安备 11010502049817号