anaconda中安装pytorch
时间: 2023-10-13 13:24:51 浏览: 109
Anaconda环境安装PyTorch文档.pdf
在anaconda中安装pytorch有以下几个步骤:
1. 打开anaconda prompt,创建一个新的虚拟环境(可选):
```
conda create --name pytorch_env
```
2. 激活虚拟环境:
```
conda activate pytorch_env
```
3. 安装pytorch:
```
conda install pytorch torchvision torchaudio -c pytorch
```
4. 安装完毕后,可以在python中测试是否成功安装:
```python
import torch
print(torch.__version__)
```
如果输出了当前安装的pytorch版本号,则安装成功。
注意事项:
- 如果要安装GPU版本的pytorch,则需要先安装CUDA和cuDNN。
- 如果遇到安装失败,可以尝试更新conda和pip,或者使用pip安装。
阅读全文