0
点赞
收藏
分享

微信扫一扫

安装GPU版本的pytorch(Windows)

芭芭蘑菇 2022-02-10 阅读 180

已安装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,安装成功!

举报

相关推荐

0 条评论