windows系统py3.8安装pytorch
时间: 2023-10-13 12:20:23 浏览: 143
pytorch安装pytorch+gpu版本安装,pytorch+cuda10.1+cudnn7.6.5安装
在Windows系统上安装PyTorch有多种方法,以下是其中一种方法:
1. 安装Anaconda
首先需要安装Anaconda,在官网上下载对应系统的Anaconda安装程序并进行安装。
2. 创建环境
打开Anaconda Prompt,创建一个PyTorch环境,输入以下命令:
```
conda create --name pytorch_env python=3.8
```
3. 激活环境
输入以下命令激活环境:
```
conda activate pytorch_env
```
4. 安装PyTorch
输入以下命令安装PyTorch:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
如果要安装GPU版本,则将`cpuonly`改为`cudatoolkit=<version>`,其中`<version>`为CUDA的版本号,例如`cudatoolkit=11.1`。
5. 测试
输入以下命令测试PyTorch是否安装成功:
```
python -c "import torch; print(torch.__version__)"
```
如果能够输出PyTorch的版本号,则安装成功。
阅读全文