pytorch环境搭建(GPU)+Anaconda+CUDA+离线安装

学习Pytorch框架,环境成为了我们第一个挫折,让我们的学习欲望“胎死腹中”是我最不愿意看到的!

1. 安装Anaconda

https://www.anaconda.com/
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

最后Finish就完事~

cmd进入刚刚指定的目录下的Lib
尝试conda -V命令查看conda版本号
在这里插入图片描述

使用conda activate激活anaconda基础环境
使用python -V查看当前anaconda的Python版本号
在这里插入图片描述

2. 安装CUDA Toolkit

先进入https://pytorch.org/查看pytorch支持的版本,因为CUDA Toolkit版本可能已经到了11.8
为了让版本匹配上,我们可以保守选择CUDA 11.6
在这里插入图片描述
然后进入https://developer.nvidia.com/cuda-toolkit-archive查看历史版本的CUDA Toolkit
在这里插入图片描述
根据自己的系统情况下载
在这里插入图片描述
安装CUDA
基本上就是默认安装!
可以看到安装的目录是C:\Users\Adminstrator\AppData\Local\Temp\CUDA
在这里插入图片描述

![在这里插入图片描述](https://img-blog.csdnimg.cn/2c7ddf3d61474525b0e1d6a6480ae881.png)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
cmd进入控制台输入nvcc -V查看自己的CUDA是否安装成功!
在这里插入图片描述

3. 离线安装PyTorch

进入https://download.pytorch.org/whl/torch_stable.html搜whl包
快捷Ctrl+F开启查找
这边建议先安装torchvision再看torch的版本号

在这里插入图片描述

参考步骤1,激活环境下的cmd,用conda create -n pytorch python=3.8命令创建一个名为pytorch的环境
在这里插入图片描述

在这里插入图片描述

创建好环境后用conda activate pytorch进入pytorch环境
在这里插入图片描述
OK,现在找到刚刚下载的包的路径,我这边是D:\torch_whl下
在这里插入图片描述
这边首先通过命令下载torchvision,pip install [touchvision你的whl完整名]
可以看到第一个pillow2.5MB,还是可以下载的,但是第二个torch包就太慢了,我们Ctrl+C取消掉,但可以看到他要下载的torch版本号是1.12.1的
在这里插入图片描述
请添加图片描述

我们再去https://download.pytorch.org/whl/torch_stable.html搜torch的whl包
请添加图片描述

下载torch包会比较慢会有点慢,可以在https://download.pytorch.org/whl/后加上它
https://download.pytorch.org/whl/cu116/torch-1.x.x%2Bcu116-cp38-cp38-win_amd64.whl注:迅雷下载会更快

安装好之后
请添加图片描述

输入命令pip install torch-1.12.1+cu116-cp38-co38-win_amd64.whl离线安装torch
请添加图片描述
看到Successfully的时候,好了!接下来用pip list查看我们安装的包,如下:
请添加图片描述

4. Pycharm环境配置,测试torch程序

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

输入代码测试一下吧~

import torch
print(torch.__version__)
print('gpu', torch.cuda.is_available())

在这里插入图片描述

自己也是一路跌跌撞撞过来的,原创不易,感谢支持~

在这里插入图片描述

  • 20
    点赞
  • 46
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值