安装python、pytorch、CUDA

安装pytorch、CUDA


写在前面:安装在win下,使用anaconda安装,不适用镜像源

  1. 安装anaconda 在anaconda官网上→Products→Individual Edition(注意Python的版本),一定注意选择加入环境变量中!!!
  2. 安装 cuda(版本),电脑界面右击→NVIDIA控制桌面→帮助→系统信息→组件→NVCUDA版本信息,看到版本信息后在cuda的网页中找到版本
    复制链接到微信,通过微信下载,下载安装
  3. 检测cuda安装:在cmd中输nvcc -V出现cuda版本信息则成功,如果找不到,则进行操作3.1
      3.1在windows搜索栏中搜索“查看高级系统设置”→环境变量→系统变量中的Path→编辑→查看是否有
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin 如果无则添加
  4. 安装Pytorch在其官网选择,注意看清楚选择的cuda的版本,复制最下面的字到“管理员权限打开的cmd”中等待下载
  5. 下载Pycharm,在官网下载community版本(专业版收费),安装好!
  6. 新建一个工程,在Pure Python→Existing interpreter→…(右侧)→Conda Environment→Interpreter→…(右侧)→conda的安装位置中Tool文件夹下python.exe文件
      6.1如何找conda安装位置:在windows搜索栏中搜索“conda”→右键文件所在位置→conda文件夹下的Tool文件夹下的python.exe
  7. 新建一个Python的 文件输入
import torch
print(torch.__version__)
print('gpu:',torch.cuda.is_available())

输出:

	1.7.1
	gpu: True

说明,pytorch,cuda均安装成功

补充python安装:
找到版本号,在官网直接输入版本号搜索后,如下下载:
安装python、pytorch、CUDA_第1张图片

你可能感兴趣的:(软件安装,cuda,pytorch)