1. 进入anaconda官网,下载anaconda,www.anaconda.com/,安装anaconda(anaconda是Windows环境下的python包管理工具)
2. 在anaconda控制台安装Python3.6(按需求选择版本)环境的包,并命名为pytorch(-n pytorch)
基础是base环境
conda create -n pytorch python=3.6
切换成pytorch环境
conda activate pytorch
安装Windows环境pytorch11.6版本(其他版本命令可前往pytorch官网查看pytorch.org/)
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
安装pytorch时,询问是否需要安装相关包,输入y
使用pip list可查看当前已安装包的情况
pip list
查看pytorch是否安装成功
如果不报错,就表示安装成功
tips:
可以切换国内镜像源安装,临时使用一次
pip install six -i https://pypi.douban.com/simple/
一劳永逸方法
pip config set global.index-url https://pypi.douban.com/simple/