PyTorch官网
可以直接去官网找下载命令https://pytorch.org/
把最下面的一行PyTorch 安装命令复制到Anaconda Prompt运行,conda和pip都可以试试,我的电脑没有GPU所以CUDA选的None,当然很完美的失败了,哪有这么简单哦,我捯饬了一天才弄好hhhh
安装三个步骤
1、创建Pytorch虚拟环境
在安装 PyTorch 之前最好先创建一个虚拟环境。原因是 Python 为不同的项目需求创建不同的虚拟环境非常常见,并且在实际项目开发中,通常会根据需求去下载各种相应框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。为 PyTorch 单独创建一个空间,这个空间与其它空间互不影响。
直接下面第三步也是ok的,但是你后面不会只用这一个包,最好是在一个独立的环境下安装,这样你再安装其他包的时候就不会互相干扰。
打开Anaconda Prompt,输入命令:
conda create -n pytorch python=3.6
意思是建一个新的环境名字叫pytorch 里面的python版本是3.6 (根据实际来)
哎嘿,出问题了,百度了一下并没有解决,依然是failed
原来需要给conda换个源,因为conda默认国外源,下载速度很慢,经常会在下载的过程中安装失败。
添加清华源:
命令行中直接使用以下命令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes
在添加清华源之后,再构建PyTorch新环境,方法依然是上面命令行,可以看到Solving environment: done
conda create -n pytorch python=3.6
2、激活新环境
会一直弹出来很多东西,最后输入 y 即可
接下来就是激活环境,使用如下命令行
conda activate pytorch
3、在新环境下安装pytorch
官网的命令行即可
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
可以看到大功告成,Successfully installed