虚拟环境pytorch如何调用
时间: 2023-10-05 09:14:43 浏览: 73
如何搭建pytorch环境
调用 PyTorch 需要先激活虚拟环境,然后在虚拟环境中安装 PyTorch 库。可以按照以下步骤来调用 PyTorch:
1. 打开终端,进入虚拟环境所在的工作目录。
2. 激活虚拟环境,命令如下:
```bash
source activate <虚拟环境名称>
```
其中,`<虚拟环境名称>` 是你创建的虚拟环境的名称。
3. 安装 PyTorch 库,命令如下:
```bash
conda install pytorch torchvision -c pytorch
```
这个命令会安装 PyTorch 和 torchvision 库。如果你只需要使用 PyTorch 库,可以省略 `torchvision`。
4. 在 Python 中导入 PyTorch 库并使用,例如:
```python
import torch
# 创建一个张量
x = torch.tensor([1, 2, 3])
print(x)
```
这个代码片段创建了一个包含整数 1、2、3 的张量,并打印出来。
记得在完成工作后,退出虚拟环境。命令如下:
```bash
source deactivate
```
这样就可以退出虚拟环境,回到系统默认的 Python 环境中。
阅读全文