anaconda 安装pytorch
如何顺利安装Anaconda中的PyTorch?不会安装的小伙伴,不妨来跟随小编今天的教程一起探索吧!
为了确保我们的Python包之间不会相互干扰,我们需要设置一个虚拟环境。这一步是非常关键的。
如何创建虚拟环境呢?只需打开终端,输入以下命令:
```bash
conda create -n torch python=3.6
```
这将创建一个名为“torch”的新的conda环境,同时指定Python版本为3.6。
接下来,激活我们刚刚创建的虚拟环境,命令如下:
```bash
activate torch
```
激活后,你会发现终端中的环境名由"base"变为了"torch"。
然后,我们来到PyTorch的官网(
```bash
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
```
将这个命令复制到终端中执行。这个命令会安装PyTorch、torchvision以及cudatoolkit。请注意,这里的cudatoolkit是用于GPU计算的库,如果你的机器没有GPU或者不需要使用GPU计算,可以跳过这个参数。
安装完成后,我们可以进入Python环境来验证PyTorch是否已经成功安装。打开Python,输入以下代码:
```python
import torch
import torchvision
```
如果没有出现报错信息,那就说明你已经成功安装了PyTorch。
以上就是小编今天的分享,希望这份教程能帮助到正在为安装PyTorch而苦恼的你。如果有任何疑问,欢迎随时向我们提问,我们会尽快为你解答。