python3.9(anaconda)+pytorch(GPU)
1.新建一个python环境
安装anaconda 5.0.1
安装名为TorchGPU的conda虚拟环境
conda create -n TorchGPU python==3.9
最终效果
conda env list
出错
在如下页面添加conda环境后,解释器列表里没反应
原因
- conda版本低,建议更新。
- conda和pycharm无法协同工作
解决
找不到就不找了,直接找到安装好的环境就行了
选择system interpreter
2.安装pytorch(GPU版本的)
我的配置:python3.9,显卡NVIDIA GeForce GTX 1050Ti
安装pythorch(GPU版本)的指令
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113
3.测试安装效果
#在TorchGPU环境下输入
import torch #不报错就算成功
torch.cuda.is_available() #输出true就算成功
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 单线程的Redis速度为什么快?
· 展开说说关于C#中ORM框架的用法!
· Pantheons:用 TypeScript 打造主流大模型对话的一站式集成库
· SQL Server 2025 AI相关能力初探
· 为什么 退出登录 或 修改密码 无法使 token 失效