刚弄完anaconda,需要配置pytorch环境。

conda create -n pytorch_gao python=3.9.12
conda activate pytorch_gao

之后,在pytorch官网上,根据自己的情况,获得一段命令:

 

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这段命令在anaconda里面,pytorch_gao环境下执行:

 

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐