anaconda 安装pytorch(GPU版本)教程

小豆苗 4月前 ⋅ 415 阅读

安装anaconda的pytorch总是安装成CPU版本,看如何切换成GPU版本。

前提是要将显卡的驱动toolkit安装好。

 

1. 首先是在anaconda prompt 里面创建一个虚拟环境

conda create -n pytorch python=3.8

 

2. 进入虚拟环境:

conda activate pytorch

 

3. 安装GPU版本的pytorch/torchaudio/torchvision

cuda10.2版本:
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch -c nvidia


cuda11.7版本
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

 

4. 验证

import  torch
torch.cuda.is_available()
print(torch.cuda.get_device_name(0))

 

最后输出显卡设备信息,即表示安装成功了。

 

 

 


全部评论: 0

    我有话说: