Pytorch 1.6.0搭建
时间: 2023-08-05 17:04:12 浏览: 134
如何搭建pytorch环境
好的,下面是在Linux系统上搭建PyTorch 1.6.0的步骤:
1. 安装Anaconda(如果已经安装可以跳过这一步)。可以从Anaconda官网下载对应版本的安装包,然后按照提示进行安装。
2. 创建一个新的conda环境。可以通过以下命令创建一个名为pytorch_env的新环境:
```
conda create -n pytorch_env python=3.7
```
3. 激活新环境。可以通过以下命令激活新环境:
```
conda activate pytorch_env
```
4. 安装PyTorch。可以通过以下命令安装CPU版本的PyTorch:
```
conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch
```
如果需要安装GPU版本的PyTorch,则可以使用以下命令:
```
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
```
5. 验证安装是否成功。可以通过在Python解释器中输入以下代码来验证是否成功安装:
```
import torch
print(torch.__version__)
```
以上就是在Linux系统上搭建PyTorch 1.6.0的步骤。
阅读全文