Windows10+GPU版 pytorch 安装

最新推荐文章于 2024-06-12 11:47:29 发布

鹿晗没我帅

最新推荐文章于 2024-06-12 11:47:29 发布

阅读量270

点赞数

分类专栏： cuda环境文章标签： python anaconda cuda windows

本文链接：https://blog.csdn.net/weixin_43942739/article/details/113688009

版权

用官网pip的命令太慢，换国内的源也不行，因为国内的源会默认给你安装cpu版本。在安装成功之前经历过很多失败，要么是安装成cpu版本；要么是安装的版本和cuda不符合；要么就是pytorch下载到一半，网速不好，直接安装失败。

这里是先将pytorch下载到自己的电脑上，然后再进行安装。

一、首先确保已经安装cuda和cudnn

这里是在确保已经安装好cuda和cudnn的情况下进行的，cuda和cudnn的相关安装教程可以自己搜一下。

这里我安装的cuda版本为10.0

二：创建一个独有的虚拟环境

1、创建 conda 环境

打开cmd命令行（快捷指令：windows+r），输入此下命令

conda create -n pytorch-gpu pip python=3.7    
# pytorch-gpu是你虚拟环境的名称，“python=3.7”python的版本号为3.7

2、激活创建的 conda 环境，进入创建的虚拟环境

conda activate pytorch-gpu # 打开该环境
conda deactivate # 退出当前环境
conda info --envs # 查看所有的虚拟环境

关注