栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch安装配置

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch安装配置

    记录一下在pytorch学习中的一些问题,先从安装配置开始,以防自己之后需要配置新环境时忘了。

1.Anaconda安装

Anaconda安装就不多说了,官网上直接下载或者清华镜像下个最新版就行。

下载过程中需要注意是否有勾选自动修改环境变量,否则需要自己去添加一下环境变量。

 2.创建虚拟环境

可以在anaconda navigator上直接创建,也可以在prompt上命令行创建。

命令行创建:

最后的

ipykernel

 可以在创建环境是就安装好,也可以不加直接创建环境。

 安装好后,查看已装环境:

conda env list

说明安装成功了。

这里我安装了好几次都失败一直无法创建环境:

查了很多原因,最后应该是镜像下载问题。我的解决路径:

1.先是安装了 清华镜像

在镜像下安装一直不成功。报错应该是http访问问题。

2.按着百度的方法试了把https改成http,在镜像末尾加上/win64/都不成功。

可以在C:C:UsersTony下找到.condarc文件,可以直接修改镜像路径,但是还是不成功,不过报错变了。

3.最后我把镜像全删了下载成功。

3.寻找本机显卡适配CUDA

此处我参考:pytorch、显卡、显卡驱动、cuda版本是如何对应的 - 简书

 先在NVIDIA官网找显卡对应算力,再在wiki上找的CUDA版本对应算力。

4.在NVIDIA官网下载对应版本的CUDA

 

 选择自定义安装。

 

全勾选,一直安装就是了。 

安装的时间比较久。。并且会闪屏几次。

 5.安装适配cuDNN

去cuDNN官网上找到cuda对应的版本下载。

 下载需要注册账号。(国外服务器访问真的是特别慢。。)(好像不注册也能下载)

下载完成后是个压缩包,解压。

把这些文件放到一个文件夹cudnn中,加入到 cuda的安装路径下。

环境变量设置,在高级系统环境设置下设置环境变量。(我是win11,其他系统也一样)

 

 

 找到path,编辑。

 

 6.安装pytorch

进入pytorch官网找到适配的版本。

将这段复制下来。

在prompt中激活之前创建的环境:(后面是环境名)

activate pytorch_gpu 

 

 然后就执行官网上复制的运行就行。

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

 也可以把-c pytorch去掉用清华镜像下载,会快一点。(摘自https://www.cnblogs.com/victorxiao/p/13512258.html)

我用conda下了好多次都有很多包下不成功,所以就用pip下,成功了。

 安装完成后依次输入下列命令行看看是否可用。

 

reference:

https://www.cnblogs.com/victorxiao/p/13512258.html

https://blog.csdn.net/a583621671/article/details/87272803

https://www.jianshu.com/p/ac70300b598b

pytorch1.9.0+cuda11.4+cudnn8.2.2+python3.9+win10+nvidia3060 安装教程 - 游戏编程 ️

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/331259.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号