在anaconda配置pytorch环境

简单记录一下在anaconda配置pytorch的步骤。

  1. 如果电脑有可用cuda的GPU,就要首先准备cuda。访问pytorch官网,查看可用的cuda版本。例如目前可适配的cuda版本为10.2以及11.3。
    在anaconda配置pytorch环境_第1张图片

  2. 安装cuda,在这里是最新的版本,可以在这里寻找对应版本,我用的是10.2。注意记下此时的安装路径。可以在anaconda prompt中输入nvcc -V确认cuda版本
    在anaconda配置pytorch环境_第2张图片

  3. 在此处安装cudnn,需要先注册账户再下载zip压缩包,注意对应版本,cuda10.2对应cudnn8.3.1
    在anaconda配置pytorch环境_第3张图片
    解压缩后将三个文件夹(\bin, \include, \lib)复制到上一步cuda的路径中,如我的路径是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

网上常见的查看cudnn版本的方法需要用cat命令,我这里查看版本是直接打开\include\cudnn_version.h查看,如图所示为8.3.1
在这里插入图片描述

  1. 下载anaconda,可以试一下自己学校可不可以用教育版,一年更新一次信息即可。
  2. 首先创建一个新环境。打开anaconda prompt,可以看到目前环境名为base,输入conda create --name myenv 可创建名为myenv的新环境。现在输入conda info --envs 可查看所有环境,如创建成功应该看到有base和myenv两个环境以及其路径。输入conda activate myenvs,以激活新环境myenv。这一步参考了conda文档
  3. 如图1回到pytorch下载,根据所装cuda版本选定,然后复制最下方conda命令并运行即开始安装。如果电脑没有GPU,可以直接选中CPU,然后复制命令并运行,这一步特别特别慢,大家也可以考虑使用镜像源。pytorch下载命令 中间下载进程停止了很多次,我ctrl+c又重新运行很多次。最后一次有如下报错,按照其路径移除对应文件后重新运行命令,最终成功安装
Downloading and Extracting Packages
pytorch-1.10.1       | 951.5 MB  | ###########################################################################9 | 100% WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash(143): Could not remove or rename C:\Users\panzh\Anaconda3\pkgs\pytorch-1.10.1-py3.9_cuda10.2_cudnn7_0\Lib\site-packages\torch\lib\cudnn64_7.dll.  Please remove this file manually (you may need to reboot to free file handles)
WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash(143): Could not remove or rename C:\Users\panzh\Anaconda3\pkgs\pytorch-1.10.1-py3.9_cuda10.2_cudnn7_0\Lib\site-packages\torch\lib\cudnn64_7.dll.  Please remove this file manually (you may need to reboot to free file handles)
pytorch-1.10.1       | 951.5 MB  | ############################################################################ | 100%

TypeError('not all arguments converted during string formatting')
  1. 检测pytorch是否可用:输入python,然后输入import torchtorch.randn(5),成功产生一个随机数tensor即为可用torch
    在这里插入图片描述

附录:
conda info --envs: show all environment
conda create --name xxx: create a new environment
conda remove --name xxx --all: remove the environment
conda list: show all packages.

输入conda install jupyter安装jupyter,使用jupyter notebook以使用,会自动在chrome中打开网页,可以打开ipynb文件或者是自行创建

你可能感兴趣的:(pytorch,深度学习,人工智能)