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就算成功

posted @   尚方咸鱼  阅读(252)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 单线程的Redis速度为什么快?
· 展开说说关于C#中ORM框架的用法!
· Pantheons:用 TypeScript 打造主流大模型对话的一站式集成库
· SQL Server 2025 AI相关能力初探
· 为什么 退出登录 或 修改密码 无法使 token 失效
点击右上角即可分享
微信分享提示