10分钟安装pytorch1.8

Alex抱着爆米花

已于 2023-11-09 11:59:57 修改

阅读量2.2w

点赞数 20

文章标签： pytorch 深度学习

于 2021-03-07 21:18:26 首次发布

本文链接：https://blog.youkuaiyun.com/qq_41080854/article/details/114498043

版权

本文档详细介绍了如何创建和激活conda虚拟环境，并指导如何在环境中安装CUDA、cuDNN以及PyTorch。提供了两种PyTorch安装方法：一是通过切换国内源安装，二是手动下载whl文件进行安装。最后，验证了PyTorch的安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

创建虚拟环境并激活

conda create -n Your_name python=3.6
conda activate Your_name

在pycharm终端输入下面命令

conda install cudatoolkit=10.2

在pycharm终端输入下面命令

conda install cudnn=7.6.5

切换到国内源
https://blog.youkuaiyun.com/weixin_34910922/article/details/116721774
推荐使用中科大的源
然后复制那个 pytorch的官网命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

torch下载界面
http://download.pytorch.org/whl/torch_stable.html
2.1.搜索torch-1.8并下载amd64.whl cp36代表python3.6版本
在这里插入图片描述

2.2.搜索torchvision-0.9.0并下载amd64.whl cp36代表python3.6版本
在这里插入图片描述

2.3.下载后放到一个文件夹里面
在这里插入图片描述

2.4 安装torch
输入pip install troch(空格自动填充)
输入pip install torchvision(空格自动填充)

pip install torch-1.8.0-cp36-cp36m-win_amd64.whl
pip install "torchvision-0.9.0+cpu-cp36-cp36m-win_amd64.whl"

import torch
print("hello pytorch {}".format(torch.__version__))
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

在这里插入图片描述