安装GPU版本的pytorch（Windows）

clevernpc

已于 2022-02-10 01:43:10 修改

阅读量1.9k

点赞数 2

文章标签： pytorch python

于 2022-02-10 01:39:57 首次发布

本文链接：https://blog.csdn.net/clevernpc/article/details/122852571

版权

已安装Python 3.8 CUDA 11.0 PyCharm2021.2.3

安装Anaconda3

误删了anaconda，按照安装anaconda步骤，安装好对应版本的anaconda后，打开【开始菜单】-->【Anaconda3(64-bit)文件夹】-->【Anaconda prompt窗口】，输入下列命令，激活pytorch环境。

conda activate pytorch

安装pytorch

进入pytorch官网，找到所需版本的安装命令，Compute Platform选择CUDA，安装的pytorch就是GPU版本。我的cuda版本是11.0，在历史版本中可找到对应版本的安装命令。

复制安装命令，在Anaconda prompt窗口中使用命令安装pytorch。

测试

在pycharm中新建一个项目文件，记得选择对应的解释器路径。

输入下列代码，运行，测试pytorch，输出了gpu：False。

import torch
print(torch.__version__)
print('gpu:',torch.cuda.is_available())

解决gpu：False

在解决gpu：False的启发下，打开cmd窗口输入下列命令，查看pytorch信息

conda list

发现pytorch仍是CPU版本，原因是之前安装的CPU版pytorch没卸载（安装的时候仔细看，会提示安装的是CPU版本，此时就不要继续安装了），在cmd窗口中使用如下命令卸载（不同的安装途径卸载命令不同，选相应的卸载命令），重新安装

#由conda安装的卸载命令
conda uninstall pytorch
conda uninstall libtorch

#由pip安装的卸载命令
pip uninstall torch

进入pycharm测试，输出gpu: True，安装成功！

关注