在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上使用。

相关推荐
moonsims6 分钟前
SKYTRAC-无人机、无人机系统和城市空中交通卫星通信 – BVLOS 和 C2 卫星通信终端和任务服务器
人工智能
云卓SKYDROID9 分钟前
无人机电压模块技术剖析
人工智能·无人机·电压·高科技·云卓科技
Codebee15 分钟前
使用Qoder 改造前端UI/UE升级改造实践:从传统界面到现代化体验的华丽蜕变
前端·人工智能
用户51914958484520 分钟前
Apache服务器自动化运维与安全加固脚本详解
人工智能·aigc
yintele25 分钟前
智能AI汽车电子行业,EMS应用相关问题
人工智能·汽车
鲸屿19532 分钟前
python之socket网络编程
开发语言·网络·python
却道天凉_好个秋33 分钟前
深度学习(四):数据集划分
人工智能·深度学习·数据集
数字冰雹36 分钟前
“图观”端渲染场景编辑器
人工智能·编辑器
里昆37 分钟前
【AI】Tensorflow在jupyterlab中运行要注意的问题
人工智能·python·tensorflow
AI视觉网奇39 分钟前
pycharm 最新版上一次编辑位置
python