PyTorch 安装的步骤
时间: 2024-06-16 17:03:40 浏览: 153
以下是PyTorch安装的步骤[^1]:
1. 找到相应版本:根据你的需求,找到适合的PyTorch版本。你可以在PyTorch的GitHub页面(https://github.com/pytorch/vision#installation)上查找版本连接,或者在PyTorch官方网站(https://pytorch.org/)上查看环境命令。
2. 下载步骤cpu版本:
2.1 选择下载版本:根据你的需求选择合适的版本,例如torch 1.12.0、torchvision 0.13.0和Python 3.7。
2.2 创建虚拟环境:使用conda命令创建一个新的虚拟环境,例如`conda create -n env_torch python==3.7`。
2.3 进入虚拟环境:使用conda命令进入新创建的虚拟环境,例如`conda activate env_torch`。
2.4 安装PyTorch:使用conda命令安装PyTorch和相关依赖,例如`conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio cpuonly -c pytorch`。
2.5 验证是否安装成功:在Python环境中导入torch模块,并打印PyTorch的版本号,例如:
```python
import torch
print(torch.__version__) # 输出安装的PyTorch版本号
```
3. 安装ipykernel(可选):如果你想在Jupyter Notebook中使用PyTorch,可以安装ipykernel,以便将虚拟环境添加到Jupyter的内核列表中。
4. 将新建的环境写入notebook的kernel中:使用命令`python -m ipykernel install --user --name=env_torch`将新建的虚拟环境添加到Jupyter的内核列表中。
5. 在虚拟环境中打开Jupyter:使用命令`jupyter notebook`在虚拟环境中打开Jupyter Notebook。
6. 选择创建的虚拟环境创建文件:在Jupyter Notebook中选择新建的虚拟环境作为内核,并创建文件。
7. 在Jupyter中测试PyTorch完成:在Jupyter Notebook中编写PyTorch代码,并运行以测试PyTorch是否成功安装。
阅读全文