WIN10安装pytorch |
您所在的位置:网站首页 › 拯救者r7000能装黑苹果么 › WIN10安装pytorch |
整个目录玩玩 文章目录 前言第一步:安装python第二步:安装Visual Stdio第三步:安装CUDA第四步:安装pytorch 如果觉得写得还行的话,就点个赞再走呗,谢谢您嘞(也欢迎提点建议) 前言上个星期折腾了deepin和TensorFlow,TensorFlow安装挺麻烦的,按官网教程还有其他博主们写的来弄,最后都没弄好(可能我太菜了又没耐心),算了,懒得折腾,最后直接一句 pip install tensorflow-gpu==1.15.0 就完事儿了,至于为啥装1.15版,因为听说该版最稳定啥的。 听说pytorch比TensorFlow好用,而且也比它好装,所以跳槽pytorch。(听说一句话就完事儿?那就让我试试 ) python肯定大家都装了吧,装了的大火就直接下一步。如果没有装的话,推荐装Anaconda,集成了很多第三方库,多的就不说了,网上一大堆,这里放一个Anaconda的清华镜像传送门,需要的小伙伴自取,注意版本号和系统,2020版的应该对应python3.8,推荐2019版的,对应python3.7,稳定点,注意在安装的时候勾选添加到环境变量,省的后面自己再装了。 装好之后,命令行输入python,进入python的命令行模式,第一行会打印出python版本信息,记住是3.7还是3.6还是3.8。 由于CUDA需要VS的支持,所以要先安装VS,2015、2017、2019的都可以,这里就略过装VS的步骤了。 pytorch-GPU版需要N卡支持,还要你的N卡支持cuda,先去NVIDIA官网下载cuda安装程序,传送门在此,进去之后,选择立即下载: torch官网有步骤,见https://pytorch.org/get-started/locally/,进去后,如下图所示,选择合适自己的选项,目前官网上给的CUDA版本最高到11.0,前面装的11.1,不过不影响。 conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch 我先试过conda install,然后在solving environment那里说一堆包有问题啥的,就没忘下走了,算了,不用conda。之前配过pip的镜像,直接pip装也一样。 pip install torch=1.7.0+cu110 torchvision=0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html 但是pip安装的话,试了几个镜像源,都找不到这几个包,不知道为啥,就像下面这样: pip install torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl torchaudio-0.7.0-cp37-none-win_amd64.whl torchvision-0.8.1+cu110-cp37-cp37m-win_amd64.whl
|
今日新闻 |
点击排行 |
|
推荐新闻 |
图片新闻 |
|
专题文章 |
CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭 |