在anaconda环境下正确快速安装pytorch,并在pycharm中实战

本文详细介绍了如何在Windows10系统中,通过Anaconda创建并激活PyTorch环境,结合Nvidia显卡的CUDA版本选择合适的PyTorch版本进行安装。过程中提到了使用清华镜像加速下载,并在PyCharm中配置该环境作为解释器,最后通过代码验证PyTorch、CUDA的正确安装与运行。
摘要由CSDN通过智能技术生成

配置:
Windows10系统、已安装Anaconda、具有Invidia显卡(任务管理器查看)
在这里插入图片描述

1、点击开始菜单,搜索anaconda prompt
在这里插入图片描述
2、进入后输入conda create -n pytorch,创建以pytorch命名的新环境
pytorch可以换成任何你喜欢的名字。
在这里插入图片描述
3、conda activate pytorch进入该环境
此时,命令行首的base将变为 pytorch
在这里插入图片描述
(重点)4、开始菜单打开cmd,输入nvidia-smi查看自己的cuda版本
这里是CUDA Version: 10.1
在这里插入图片描述
(重点)5、进入pytorch官网,ctr+F搜索10.1找到适配版本
10.1换成任意第4步中你的电脑得到的版本
官网链接:https://pytorch.org/get-started/previous-versions/
这是关键,否则装上了也会因为版本适配问题而无法使用。在这里插入图片描述
6、库较大先别着急下载,回到anaconda prompt,添加清华镜像
四行命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

7、添加之后,拷贝pytorch官网这句话,删去后面的’-c pytorch’
'-c pytorch’的意思是让你从官网下载,国内被墙,下载慢。
只输入这一部分就行:

conda install pytorch== 1.7.0 torchvision== 0.8.0 torchaudio== 0.7.0 cudatoolkit = 10.1

之后开始下载,有y按y。
如果最终报错:Downloaded bytes did not match Content-Length。
可能是你的网络原因也可能是镜像源出现问题,检查并非网络问题后,可以回到第5步,在官网中换一个pytorch版本下载。
8、打开pycharm,创建项目
项目自行创建,不做演示。
接着打开file->settings
在这里插入图片描述
如下图,点击右侧齿轮按钮,选Add
在这里插入图片描述
左侧选择Conda Environment,勾选Existing environment
第一个Interpreter的框,选择步骤2中,安装pytorch环境的那个位置,里面有python.exe文件
之后Apply、确认即可
在这里插入图片描述
9、实战检验
输入代码:

import torch
import torch.nn as nn
from torch.autograd import Variable

print(torch.cuda.is_available())

正确运行,返回true
在这里插入图片描述

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值