确保Pytorch安装成功的一份不完全踩坑指南(windows10)

您所在的位置:网站首页 forge安装不完整怎么办 确保Pytorch安装成功的一份不完全踩坑指南(windows10)

确保Pytorch安装成功的一份不完全踩坑指南(windows10)

2024-07-07 18:05| 来源: 网络整理| 查看: 265

踩了不少坑,给大家po一个合理的安装顺序:

安装Anaconda

最好先在anaconda prompt里new一个conda的python3.6虚拟环境(方法去别的文章搜一下),再整理一下conda的pkg文件夹确保你以前没有安装过跟torch以及cuda toolkit相关的包。如果有的话,建议重装一下anaconda,或者clean一下list里的包。因为这个我配环境配了好久,出了很多问题。

CUDA Toolkit

先更新你的显卡驱动,现在nvidia的最新驱动好像是集合了,适配多张显卡的,更新完毕后。 打开prompt,查看CUDA适配的最高版本

nvidia-smi

如图,低于496.13以及11.5的cuda版本都是可以安装的。

然后先去pytorch官网上找你要安装的torch版本 如图,版本是稳定版的torch1.10。CUDA是10.2,最好严格按照10.2安装CUDA,11.3也一样。先复制下面的命令行备用。 在这里插入图片描述 这儿贴一个CUDAtoolkit的旧版本列表,如果是10.2,找到10.2下载并安装。(安装时最好只自定义安装CUDA,把显卡驱动部分取消勾选)Archive of Previous CUDA Releases 安装完毕后下载CUDNN这里选最新的适配CUDA10.2的CUDNN就可以,下载后解压复制进CUDA的文件夹里覆盖就算安装成功。

conda prompt里输入查看cuda安装情况

nvcc -V

如图可以查看安装情况 在这里插入图片描述

Pytorch安装

conda激活一下你new的虚拟环境,然后复制上面备用的torch安装命令等待包下载安装即可,速度慢的话网上搜一个清华源或者挂代理,另外确保你的网络连接比较稳定,不然安装容易出问题。

包下载完以后好像会有三个确认,我之前因为conda的包列表路径不干净,cudatoolkit安装老是出问题,搜了不少文章有用pip wheel强制安装的,但是我搜不到对应的轮子文件,也有说是版本不对应的,试了很多次,只有重装conda这个方法合适。

安装完毕后进入python

python import torch torch.cuda.is_available()

显示true则torch安装成功 在这里插入图片描述 有别的问题可以评论区问我。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3