您当前的位置: 首页 >  pytorch

耐心的小黑

暂无认证

  • 3浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

pytorch学习笔记一:简述在Anaconda下安装pytorch1.4的步骤

耐心的小黑 发布时间:2020-02-12 11:38:24 ,浏览量:3

  1. 首先在官网下载安装好Anaconda,目前pytorch支持python3.5-3.7,所以应该下载所需版本

  2. 安装CUDA传送门和CUDNN传送门,注意一定要和目前的pytorch支持的版本匹配,比如CUDA10.1,cuDNNv7.6.5。具体安装过程,参考此处传送门

  3. 在Anaconda中创建虚拟环境,在开始菜单中打开 Anaconda Prompt,然后在命令行输入conda create -n pytorch python=3.x以创建虚拟环境

  4. 到pytorch官网点击此处获取安装链接。下面是加了清华镜像源的链接,这样下载更稳定,也更快;pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.4.0 torchvision ==0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

  5. 然后激活刚才创建的虚拟环境,使用conda activate pytorch,将上述的下载链接输入到命令行,等待安装完成

  6. 最后使用pip,安装各种需要的包即可,注意是在虚拟环境下使用pip

关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.0388s