anaconda 安装pytorch（GPU版本）教程

安装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))

 

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

 

 

 

注意：本文归作者所有，未经作者允许，不得转载
原文地址： http://blog.wsmee.com/post/192
版权声明：非商用-非衍生-保持署名| Creative Commons BY-NC-ND 3.0