pycharm+pytorch环境搭建

编程小工

已于 2024-04-28 22:32:33 修改

阅读量340

点赞数 3

文章标签： python pytorch pycharm

于 2024-04-28 21:44:30 首次发布

本文链接：https://blog.csdn.net/qq_22250743/article/details/138292281

版权

python的解释器

将python语言翻译成机器指令语言

各环境间相互独立

Anaconda是为了方便使用python而建立的一个软件包，其包含常用的250多个工具包，多版本的python解释器和强大的虚拟环境管理工具，所以anaconda得名python全家桶，anaconda可以使安装运行和升级环境变得更简单，因此推荐安装使用

安装结束后：

验证是否安装成功，cmd执行：

安装完成后进入安装目录，验证

需要注册账号下载

将cunn的压缩包解压，将三个文件夹复制，粘贴到cuda安装目录下

安装完成后运行demo程序进行验证

设备查询

安装pytorch

可以从

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 链接的网站中直接下载

https://download.pytorch.org/whl/cu121

下载torch

下载torchvision

在pycharm的consle中创建

 conda create -n dicom_gpu python=3.11.7

激活环境

进入torch及torchversion的安装目录

安装torch

 pip install "torch-2.1.0+cu121-cp311-cp311-win_amd64.whl"

安装torchversion

 pip install "torchvision-0.16.0+cu121-cp311-cp311-win_amd64.whl"

设置pycharm的python解释器

运行代码验证

import torch
from torchvision import models

print("Hello PyTorch{}".format(torch.__version__))

print(torch.cuda.is_available())

验证成功:

关注