如何安装PyTorch?
时间: 2024-08-04 14:00:53 浏览: 158
要安装PyTorch,按照以下步骤操作:
1. **安装 Anaconda**:
官方网站下载最新版Anaconda,安装完成后,在命令行中启动Anaconda Prompt并确认显示`(base)`表示安装成功。
2. **创建虚拟环境**:
使用`conda create`命令创建一个名为`pytorch`的新环境,指定Python版本:
```shell
conda create -n pytorch python=3.6
```
3. **激活虚拟环境**:
Linux下:
```shell
source activate pytorch
```
Windows下:
```shell
conda activate pytorch
```
4. **检查现有包**:
查看当前环境中已安装的包:
```shell
pip list
```
5. **安装 PyTorch**:
直接从PyTorch官网获取最新的安装指令,如需CUDA支持,添加相应版本:
```shell
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
6. **验证安装**:
检查是否安装成功,通过导入`torch`模块并检查GPU可用性:
```python
import torch
if torch.cuda.is_available():
print("PyTorch and CUDA installed successfully.")
```
如果执行这些步骤后,一切正常,那么你就已经成功安装了PyTorch。记得在后续项目中激活该环境以使用它。
阅读全文