当前位置: 首页> 娱乐> 八卦 > Pytorch和Tensorflow安装【Win和Linux】

Pytorch和Tensorflow安装【Win和Linux】

时间:2025/7/11 15:34:18来源:https://blog.csdn.net/qq_38758371/article/details/140036031 浏览次数:0次

Ubuntu/win安装Pytorch和Tensorflow

说明:

这两种框架的搭建,均基于Anaconda进行搭建。先在系统中安装Anaconda软件。

一、Pytorch的搭建

windows安装

(1)搭建参考官网给的命令,pytorch官网

(2)下载地址:https://download.pytorch.org/whl/torch_stable.html

从上述地址自行下载对应的版本(要求符合python的版本,cWuda 的版本信息)

比如我的CUDA是11.2 ,python是3.7版本,在Windows系统下

那么下载cu111/torch-1.8.1%2Bcu111-cp36-cp36m-win_amd64.whl

下载完成后,在cmd界面跳转到文件所在的目录下面,然后执行安装

pip install cu111/torch-1.8.1%2Bcu111-cp36-cp36m-win_amd64.whl

Linux安装

(1)创建虚拟环境

(2)安装pytorch

我们改变策略 。记录好需要安装的包,直接用pip从豆瓣源下载。根据官方命令,我们需要安装的是torch1.8.1版本,torchvision0.9.1版本,torchaudio==0.8.1版本。

安装torch==1.8.1

pip install torch==1.8.1 -i https://pypi.doubanio.com/simple --trusted-host pypi.doubanio.com

安装torchvision==0.9.1

pip install torchvision==0.9.1 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com

安装torchaudio==0.8.1

pip install torchaudio==0.8.1 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com

关于版本说明

torchtorchvision对应关系,来源:官方Repo: vision

torchtorchvisionpython
main / nightlymain / nightly>=3.8, <=3.10
2.0.00.15.1>=3.8, <=3.10
1.13.00.14.0>=3.7.2, <=3.10
1.12.00.13.0>=3.7, <=3.10
1.11.00.12.3>=3.7, <=3.10
1.10.20.11.3>=3.6, <=3.9
1.10.10.11.2>=3.6, <=3.9
1.10.00.11.1>=3.6, <=3.9
1.9.10.10.1>=3.6, <=3.9
1.9.00.10.0>=3.6, <=3.9
1.8.20.9.2>=3.6, <=3.9
1.8.10.9.1>=3.6, <=3.9
1.8.00.9.0>=3.6, <=3.9
1.7.10.8.2>=3.6, <=3.9
1.7.00.8.1>=3.6, <=3.8
1.7.0
关键字:Pytorch和Tensorflow安装【Win和Linux】

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: