pytorch安装
时间: 2023-07-03 13:28:22 浏览: 138
PyTorch是一个基于Python的科学计算包,主要针对深度学习,提供了很多高级的函数和算法。安装PyTorch的方法如下:
1. 安装Anaconda
首先需要安装Anaconda,可以从Anaconda官网下载对应操作系统的版本,并按照安装指南安装好Anaconda。
2. 创建虚拟环境
在Anaconda Prompt中输入以下命令,创建一个名为"pytorch"的虚拟环境:
```
conda create --name pytorch python=3.7
```
3. 激活虚拟环境
在Anaconda Prompt中输入以下命令,激活刚才创建的虚拟环境:
```
conda activate pytorch
```
4. 安装PyTorch
在Anaconda Prompt中输入以下命令,安装PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
其中,cudatoolkit=11.1表示使用CUDA Toolkit的版本,此处选择的是11.1版本,如果你的机器上没有安装CUDA,则可以不加此参数。
5. 验证安装
在Anaconda Prompt中输入以下命令,验证PyTorch是否安装成功:
```
python -c "import torch; print(torch.__version__)"
```
如果输出了PyTorch的版本号,则说明安装成功。
注意:如果你的机器没有NVIDIA的GPU,则需要安装CPU版本的PyTorch,命令如下:
```
conda install pytorch-cpu torchvision torchaudio cpuonly -c pytorch
```
阅读全文