在win10上安装pytorch-gpu版本2

安装anaconda即下载了python,还可以创建虚拟环境。

目录

1.1 anaconda安装

1.2 pytorch-gpu安装


1.1 Anaconda安装

anaconda的安装请看我之前发的tensoflow-gpu安装,里面有详细的安装过程,这里不做重复描述,传送门

1.2 pytorch-gpu安装

anconda下载完成后,打开anaconda prompt,输入

conda create -n pytorch-gpu python=3.7 创建pytorch用的虚拟环境,当然你们虚拟环境也可以不用命名成pytorch-gpu,创建好后,在pytorch官网找到对应的安装包就行下载,我这里直接

复制代码
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

如果安装太慢就需要加上镜像源了

复制代码
永久生效 (pip config set global.index-url+国内镜像地址)
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/

# 添加Anaconda的TUNA镜像
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 --set show_channel_urls yes

输入这个命令就行下载,如果速度慢可以使用镜像源进行下载,

下载完成之后,在命令行先输入python

再输入

复制代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())

结果成功即为安装成功。

然后在pycharm里面配置anconda的虚拟环境打开file-setting-Python Interpreter-Add添加anaconda的环境

这样就配置好pytorch-gpu版本,并在pycharm上使用。

相关推荐
星期天要睡觉25 分钟前
自然语言处理(NLP)——自然语言处理原理、发展历程、核心技术
人工智能·自然语言处理
低音钢琴29 分钟前
【人工智能系列:机器学习学习和进阶01】机器学习初学者指南:理解核心算法与应用
人工智能·算法·机器学习
飞翔的佩奇2 小时前
【完整源码+数据集+部署教程】【天线&水】舰船战舰检测与分类图像分割系统源码&数据集全套:改进yolo11-repvit
前端·python·yolo·计算机视觉·数据集·yolo11·舰船战舰检测与分类图像分割系统
大千AI助手2 小时前
Hoeffding树:数据流挖掘中的高效分类算法详解
人工智能·机器学习·分类·数据挖掘·流数据··hoeffding树
新知图书2 小时前
大模型微调定义与分类
人工智能·大模型应用开发·大模型应用
山烛2 小时前
一文读懂YOLOv4:目标检测领域的技术融合与性能突破
人工智能·yolo·目标检测·计算机视觉·yolov4
大千AI助手2 小时前
独热编码:分类数据处理的基石技术
人工智能·机器学习·分类·数据挖掘·特征工程·one-hot·独热编码
钱彬 (Qian Bin)2 小时前
项目实践4—全球证件智能识别系统(Qt客户端开发+FastAPI后端人工智能服务开发)
人工智能·qt·fastapi
钱彬 (Qian Bin)2 小时前
项目实践3—全球证件智能识别系统(Qt客户端开发+FastAPI后端人工智能服务开发)
人工智能·qt·fastapi
Microsoft Word2 小时前
向量数据库与RAG
数据库·人工智能·向量数据库·rag