建站优化

当前位置:

anaconda 安装pytorch

浏览量:57次

anaconda 安装pytorch

anaconda怎么安装pytorch呢?不知道的小伙伴来看看阳泰小编今天的分享吧!

1、设置虚拟环境。之所以要设置虚拟环境,是因为这个 python里面的包有很多个,有些个包可以会存在相互干扰的问题,所以用这个虚拟环境来把他们来隔绝开。

创建虚拟环境:conda create -n torch python=3.6

激活虚拟环境:activate torch,

当激活torch环境之后,左边的环境就由base变为了torch。

2、去Pytorch的官网(https://pytorch.org/get-started/locally/)下找到PyTorch安装需要使用的命令 。

将代码conda install pytorch torchvision cudatoolkit=10.1 -c pytorch粘到命令行里面。

进入python环境,然后输入import torch 和import torchvision验证,如果没有报错的话,就说明安装已经成功了。

[声明]本网转载网络媒体稿件是为了传播更多的信息,此类稿件不代表本网观点,本网不承担此类稿件侵权行为的连带责任。故此,如果您发现本网站的内容侵犯了您的版权,请您的相关内容发至此邮箱【779898168@qq.com】,我们在确认后,会立即删除,保证您的版权。