文章目录

  • 1.先安装anaconda
  • 2.cuda 10.2及cudnn的安装
  • 2.1cuda 10.2下载及安装
  • 2.2 cudnn的安装
  • 3安装Pytorch
  • 3.1配置下Anaconda的下载源
  • 3.2 创建名字为pytorch的虚拟环境
  • 3.3激活环境
  • 3.4 验证一下
  • 4 jupyter 创建基于pytorch这个虚拟环境的文件
  • 4.1
  • 4.2修改anaconda jupyter默认路径


1.先安装anaconda

官网:https://www.anaconda.com/products/individual

cuda和pytorch混编 conda pytorch cuda_python


cuda和pytorch混编 conda pytorch cuda_cuda和pytorch混编_02


虽然写着python3.8,是向下兼容的,下载就行。

安装过程就略了。

安装完检查Anaconda的环境变量。没有把这俩添加进去。

cuda和pytorch混编 conda pytorch cuda_anaconda_03

2.cuda 10.2及cudnn的安装

2.1cuda 10.2下载及安装

cuda和pytorch混编 conda pytorch cuda_cuda和pytorch混编_04


按照官网给出的提示,安装稳定版Pytorch 1.60,需要cuda版本10.2 ,下载地址:https://developer.nvidia.com/cuda-toolkit-archive

cuda和pytorch混编 conda pytorch cuda_python_05


cuda和pytorch混编 conda pytorch cuda_python_06


大小为2.6G。请默认安装,大家安装位置都一致。

2.2 cudnn的安装

https://developer.nvidia.com/rdp/cudnn-download

cuda和pytorch混编 conda pytorch cuda_cuda和pytorch混编_07


需要注册账号,也不麻烦,就是填几个项。下载是个压缩包。

  • 把压缩包中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目录下对应目录中
  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64\cupti64_102.dll 拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
  • 再配置下环境变量:
  • 确保已经将上面4个path添加到系统变量中!

3安装Pytorch

3.1配置下Anaconda的下载源

这个源跟python的pip下载源一样的。配置文件位置: C:\Users\your_name下名为.condarc文件。

cuda和pytorch混编 conda pytorch cuda_cuda和pytorch混编_08


cuda和pytorch混编 conda pytorch cuda_cuda_09

一般就用清华的:这是我的。

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
ssl_verify: true
show_channel_urls: true
auto_activate_base: false
channel_priority: flexible

3.2 创建名字为pytorch的虚拟环境

Anaconda Prompt 终端输入(不是windows的cmd): conda create -n pytorch python=3.6就创建了名字为 pytorch的虚拟环境,(这里-n相当于 --name),这里一般会下载python3.6和其他几个基础包。

cuda和pytorch混编 conda pytorch cuda_虚拟环境_10

3.3激活环境

conda activate pytorch

conda install pytorch torchvision cudatoolkit=10.2

这里注意不要官方命令原版的结尾的-c pytorch

cuda和pytorch混编 conda pytorch cuda_python_11


cuda和pytorch混编 conda pytorch cuda_anaconda_12

cuda和pytorch混编 conda pytorch cuda_cuda和pytorch混编_13


这样就安装好了。

3.4 验证一下

cuda和pytorch混编 conda pytorch cuda_python_14

4 jupyter 创建基于pytorch这个虚拟环境的文件

4.1

anaconda安装好默认环境叫base,现在创建了一个pytorch的环境,但是我们打开

cuda和pytorch混编 conda pytorch cuda_虚拟环境_15


里只能基于base创建jupyter文件。我想在打开的jupyter (Anaconda)创建其他虚拟环境的:

  • 先在base默认环境中:
    输入conda install ipykernel安装ipykernel,或可以用conda list查看是否已经安装了。
  • 在虚拟环境下创建kernel文件:下面这个pytorch可以换成其他环境名。
    先激活该虚拟环境:
    conda activate pytorch 然后:
    conda install -n pytorch ipykernel
  • 继续在该虚拟环境中,写入notebook的kernel中
    python -m ipykernel install --user --name 环境名称 --display-name "在jupyter中显示的环境名称" 我的命令是:python -m ipykernel install --user --name pytorch --display-name "pytorch"
  • cuda和pytorch混编 conda pytorch cuda_python_16

  • 可以验证,打印版本:
  • cuda和pytorch混编 conda pytorch cuda_虚拟环境_17


4.2修改anaconda jupyter默认路径

jupyter notebook --generate-config

cuda和pytorch混编 conda pytorch cuda_虚拟环境_18


打开该文件,找到notebook_dir

#c.NotebookApp.notebook_dir = ''改为:前面的注释#要去掉。

c.NotebookApp.notebook_dir = 'C:\\Users\\liufe\\Desktop\\Jupyter Projects'

cuda和pytorch混编 conda pytorch cuda_anaconda_19

第4节参考: