安装了一上午终于成功了!安装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,替换成我们需要的库名就好啦!

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐