Pytorch环境安装

必备组件:

  • Anaconda3-2021.11-Windows-x86_64
  • Python3.9
  • cuda_11.1.0_456.43_win10
  • cudnn-windows-x86_64-8.3.1.22_cuda11.5-archive
  • torch-1.10.0+cu113-cp39-cp39-win_amd64
  • torchvision-0.11.1+cu113-cp39-cp39-win_amd64
  • 编译器,PyCharm

(torch下载的版本要与本地python的版本对应,不然大概率装不上)

步骤:

  1. 安装Anaconda,目录可以自定,要记得在哪,安装时注意勾选上添加到系统环境变量的选项;
  2. 安装CUDA,尽量选择默认目录;
  3. 安装CUDAnn,是个压缩包,解压后复制其中的文件夹到上一步CUDA的安装目录下粘贴覆盖;
  4. 打开PyCharm,新建一个工程,解释器选择对应的python版本,这里使用的是3.9;
  5. 工程内新建一个文件,键入如下代码,torch安装之前,import torch应该是标红线报错的;在这里插入图片描述
  6. 点击左下角Terminal命令窗口,键入:conda create -n pythorch_gpu python=3.9
    创建gpu运算环境,建立过程中的[y/n]全部键入y;
    在这里插入图片描述
  7. 激活虚拟环境,命令:conda activate pytorch_gpu;
  8. 激活后,cd到torch与torchversion所在的目录,使用pip install命令进行安装;
  9. 安装完成后,在当前工程下,更换解释器为刚刚用conda建立的虚拟环境:在这里插入图片描述
  10. 点击右上角的齿轮,点add,添加新建的解释器(刚刚的torch就安装在这里);在这里插入图片描述
    左侧选择conda,右侧点击指定环境,然后找到conda的安装目录,在anaconda>envs目录下找到python.exe, 点击ok>ok;
  11. 最后运行步骤5中的代码,得到如下结果即为安装成功:在这里插入图片描述
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值