首先创建个虚拟环境: conda create -n 虚拟环境名字 python=指定python版本

进入pytorch的官网,直接复制命令即可

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FxuFrfxO-1609146629132)(images/image-20201224112811344.png)]

pytorich官网是国外源,对于不能翻墙的小伙伴可能不太友好,所以还有第二种方式安装。

用pip安装whl文件

pytorch的whl文件下载地址:https://download.pytorch.org/whl/torch_stable.html

根据自己的需求,去上述网址中直接下载对应版本的whl文件即可,cpu和gpu版本都有。

安装指令为:pip install torch-1.6.0+cu101-cp36-cp36m-linux_x86_64.whl

我这里下载的就是linux版本的torch1.6.0,对应的cuda10.1版本,python3.6版本

如图所示:
在这里插入图片描述

图片重点是演示操作步骤和安装命令,所以名字和系统跟上述文字部分不符合,但是道理都是一样的。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐