PyTorch 深度学习虚拟环境安装与配置 GPU 版

什么是虚拟环境?

在 Anaconda 中,虚拟环境允许你为不同的项目创建隔离的 Python 环境,这样可以避免不同项目之间的依赖冲突。

使用虚拟环境是一个很好的实践,特别是当你在开发多个项目,或者需要不同版本的库时。这样可以确保项目的依赖清晰,并且易于管理。

安装虚拟环境

查看所有虚拟环境

复制代码
conda env list

创建一个虚拟环境

利用 conda create 指令 创建新的虚拟环境(注意:虚拟环境的名字取得一定要有意义)

复制代码
conda create --n 虚拟环境名字 python=版本

删除虚拟环境:

复制代码
conda remove --n 虚拟环境名字 --all

激活虚拟环境

复制代码
conda activate 虚拟环境名字

安装Pytorch

在上一步创建的虚拟环境中安装PyTorch(安装PyTorch,需要安装pytorch, torchvision,

torchaudio三个包)

进入到pytorch的官网:

Start Locally | PyTorch Start Locallyhttps://pytorch.org/get-started/locally/选择对应的参数后,复制下面的指令到命令行运行

验证安装:

1. 激活对应的虚拟环境(你安装Pytorch的虚拟环境)conda activate 虚拟环境名
2. 输入conda list,看有没有pytorch或者torch
3. 输入python
4. 输入 import torch
5. 输入 torch.cuda.is_available()
6. 如果显示True,就说明我们这个PyTorch安装成功了

安装完成后,你可以在 Python 中验证 PyTorch 是否安装成功:

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

给下载的项目设置合适的虚拟环境

  1. 利用PyCharm打开项目:File->Open

  2. 配置对应的虚拟环境:File->Setting->Project Interpreter->Python解释器,然后选择你在命令行创建的对应的虚拟环境

  1. 直接运行代码,右键->run

  2. 如果提示某些包没有发现,大家可以用Conda/pip install 包名(注:如果安装错误,利用搜索引擎找找原因 -- 包名不对,通道不对,或者其他原因)

  3. 可选-最好把requirements.txt文件的内容当作参考。

有选择性的使用 :(1.打开命令行进入到项目文件的位置,命令:cd+项目文件路径

2.输入指令:pip install -r requirements.txt即可,就会自动安装)

相关推荐
OceanBase数据库官方博客几秒前
OceanBase seekdb-cli:专为 AI Agent 设计的数据库接口
数据库·人工智能·oceanbase
MatrixOrigin17 分钟前
什么是AI Native的组织,它该具备什么样的特点
人工智能·ai·opc
xiaoxiaoxiaolll18 分钟前
金属结构疲劳寿命预测与健康监测技术
人工智能·算法·机器学习
AI视觉网奇32 分钟前
3d部件拆分PartCrafter
人工智能·大模型
Ether IC Verifier32 分钟前
CPU/GPU/NPU/DPU功能详解与AI应用分析
网络·人工智能·网络协议·tcp/ip·计算机网络·dpu
2601_9494999439 分钟前
芯瑞科技800G VR8 OSFP光模块,解决智算中心高速互连四大难题
人工智能·科技
清风lsq1 小时前
大模型-vllm 投机解码实现
人工智能·vllm·大模型推理
2601_958352901 小时前
拆解 EN-46:一块 15mA 的 DSP 芯片如何实现 50dB 降噪
人工智能·语音识别·信号处理·嵌入式开发·音频降噪·双麦波束成形·硬件拆解
Mangguo52081 小时前
GPU利用率提升软件横向对比评测:NVIDIA Run:ai、VMware Bitfusion、博云ACE与阿里云PAI灵骏
人工智能·阿里云·云计算
摄影图1 小时前
蓝色光效科技背景图片素材 多场景设计
人工智能·科技·aigc·贴图·插画