【无标题】cuda11.7+python3.7+pytorch GPU

anaconda

官网下载anaconda:链接: https://www.anaconda.com/
注意在下下载时,选择PATH加入路径

【无标题】cuda11.7+python3.7+pytorch GPU
安装完检查一下
【无标题】cuda11.7+python3.7+pytorch GPU

如果不对,检查环境变量

【无标题】cuda11.7+python3.7+pytorch GPU

; cuda

现在的pytorch支持的最高版本是cuda 11.6,安装11.7后,系统自动安装了cpu版本的pytorch,我试了anaconda里安装和命令行安装都不行,之前试过从网页上直接下载下来pytorch-gpu版本的包,然后命令行用conda安装,我记得是可以的,这次没有试。
特意去网上找了 驱动与cuda版本对应关系,网址:链接: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html,很多教程写的都是查电脑显示,建议安装的cuda版本,其实只要从表格中看,驱动对应的可以下载的cuda版本,都是可以的。 可以先去看看要安装的python包,现在最高支持的cuda版本,再选择cuda版本,pytorch现在最高支持11.6

【无标题】cuda11.7+python3.7+pytorch GPU
【无标题】cuda11.7+python3.7+pytorch GPU

先下载 迅雷,cuda官网:
11.7:链接:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
11.6:链接:https://developer.nvidia.com/cuda-11-6-2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
下载很慢

【无标题】cuda11.7+python3.7+pytorch GPU

右键download那个按钮,复制链接,粘贴到迅雷中下载。
安装时的小坑:解压路径和安装路径不能选择同一个,否则,安装完,解压路径会被自动删除,安装的东西也没了。
看一下是否安装完成:nvcc -V

【无标题】cuda11.7+python3.7+pytorch GPU

如果没安装好,检查一下环境变量

【无标题】cuda11.7+python3.7+pytorch GPU

; cudnn(其实不装cudnn也可以用)

网址:链接:https://developer.nvidia.com/rdp/cudnn-archive,需要注册,qq邮箱一直收不到验证码,后来改成google邮箱,秒收到。
cudnn直接在网站上下载,用迅雷会变慢。根据cuda版本选择合适的版本就行。下面是我下载的版本

【无标题】cuda11.7+python3.7+pytorch GPU
解压完成后,将三个文件夹下的所有文件拷贝至CUDA安装目录下对应的文件夹中:
【无标题】cuda11.7+python3.7+pytorch GPU

; pytorch

命令行安装

conda 里面新建环境,可以去网上找找,我是在anaconda里面创建的,可参考下面。
创建环境并安装:

conda create -n 你的环境名称 python=3.7
activate 你的环境名称
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

注意看一下是不是gpu版本

【无标题】cuda11.7+python3.7+pytorch GPU
我注意到安装时,这里会安装coda,所以可能直接用上面命令,不安装cuda也是可以的
【无标题】cuda11.7+python3.7+pytorch GPU

anaconda安装

  • 换源
    win+R:%HOMEPATH%
    【无标题】cuda11.7+python3.7+pytorch GPU

用记事本打开:

【无标题】cuda11.7+python3.7+pytorch GPU

粘贴 channnel,保存。试着在网上找了很多源,下面这个是有3.7,也可以顺利创建环境

channels:
  - http://mirrors.aliyun.com/anaconda/pkgs/main
  - http://mirrors.aliyun.com/anaconda/pkgs/r
  - http://mirrors.aliyun.com/anaconda/pkgs/msys2
show_channel_urls: true
ssl_verify: false

  • 新建环境:(关掉梯子!!!!)

【无标题】cuda11.7+python3.7+pytorch GPU
这次安装,在包里面搜不到torch-gpu,应该是这个源里面没有,可以去找找,如果能搜到,安装就很简单。

下载好安装包进行安装

等下次安装有机会,我找个网页,把这一部分补上

检查是否安装gpu版本

python
>>> import torch
>>> print(torch.cuda.is_available())

【无标题】cuda11.7+python3.7+pytorch GPU

Original: https://blog.csdn.net/qq_43206709/article/details/126601453
Author: qq_43206709
Title: 【无标题】cuda11.7+python3.7+pytorch GPU

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706224/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球