Ananconda下安装pytorch(GPU),亲测成功(清华源镜像快速下载)
安装了一上午终于成功了!创建Pytorch虚拟环境打开Anaconda Prompt,输入下列命令:conda create -n pytorch python=3.8启用创建的环境:conda activate pytorch点击此链接https://pytorch.org/get-started/locally/,进入pytorch官网,根据自己的配置,生成对应的pytorch下载指令。选择下
安装了一上午终于成功了!安装torch需要对应cuda和python版本,否则会出问题。
点击此链接https://pytorch.org/get-started/locally/
进入官网,点击Previous Pytorch Versions,会看到所有版本的torch的命令。
以其中一个版本为例,可以看到对应不同的cuda,此时需要查看自己电脑支持的cuda(一般不是太高的版本都能支持),然后再查看自己要运行的代码是否有标明运行环境的cuda和torch,需按照要求来,否则会出现精度不一致的问题。
找到需要的cuda版本之后,需要先去下载cuda和cuDNN。下好之后,再来找对应的命令安装torch。
关于这个问题,有个博主写的非常详细,可参考!
一看就懂的 CUDA安装教程及Pytorch GPU版本安装教程_python安装cuda_Evan_qin_yi_quan的博客-CSDN博客
此外还需要确定torch对应的python版本。(这个直接百度搜就行)
创建Pytorch虚拟环境
打开Anaconda Prompt,输入下列命令:
conda create -n pytorch python=3.8
启用创建的环境:
conda activate pytorch
选择下载的方式,由于conda要配置源,对萌新不太友好,所以我们选择pip。
直接Run this command会很慢(前期我直接运行了好几次,以失败告终 哭泣泣~~~),我们只需要切换到国内源,就会体验飞一般的感觉!
-i https://pypi.tuna.tsinghua.edu.cn/simple some-package
最后我们运行的命令为:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
输出结果为True,就是成功了!!!
当然,我们安装其他库是也可以使用这个命令进行快速下载!!!
pip install xxx -i https://pypi.tuna.tsinghua.edu.cn/simple
如果这个不成功,就再试试这个!
pip install 包名 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
把xxx,替换成我们需要的库名就好啦!
更多推荐
所有评论(0)