Anaconda下pytorch的cpu版和gpu版安装
首先创建个虚拟环境: conda create -n 虚拟环境名字 python=指定python版本进入pytorch的官网,直接复制命令即可pytorich官网是国外源,对于不能翻墙的小伙伴可能不太友好,所以还有第二种方式安装。用pip安装whl文件pytorch的whl文件下载地址:https://download.pytorch.org/whl/torch_stable.html根据自己的
·
首先创建个虚拟环境: conda create -n 虚拟环境名字 python=指定python版本
进入pytorch的官网,直接复制命令即可
![[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FxuFrfxO-1609146629132)(images/image-20201224112811344.png)]](https://i-blog.csdnimg.cn/blog_migrate/1e357a1561b85ac36f82da221f335296.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版本
如图所示:
图片重点是演示操作步骤和安装命令,所以名字和系统跟上述文字部分不符合,但是道理都是一样的。
更多推荐




所有评论(0)