确保Pytorch安装成功的一份不完全踩坑指南(windows10) |
您所在的位置:网站首页 › forge安装不完整怎么办 › 确保Pytorch安装成功的一份不完全踩坑指南(windows10) |
踩了不少坑,给大家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也一样。先复制下面的命令行备用。 conda prompt里输入查看cuda安装情况 nvcc -V如图可以查看安装情况 conda激活一下你new的虚拟环境,然后复制上面备用的torch安装命令等待包下载安装即可,速度慢的话网上搜一个清华源或者挂代理,另外确保你的网络连接比较稳定,不然安装容易出问题。 包下载完以后好像会有三个确认,我之前因为conda的包列表路径不干净,cudatoolkit安装老是出问题,搜了不少文章有用pip wheel强制安装的,但是我搜不到对应的轮子文件,也有说是版本不对应的,试了很多次,只有重装conda这个方法合适。 安装完毕后进入python python import torch torch.cuda.is_available()显示true则torch安装成功 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |