安装pytouch，cuda，实现gpu计算

车前猛跑

于 2024-06-06 10:20:12 发布

阅读量579

点赞数 2

分类专栏：机器学习基础文章标签： python pytorch

本文链接：https://blog.csdn.net/cin_ie/article/details/137379411

版权

17 篇文章

订阅专栏

# 创建虚拟环境
conda create -n pytorch_gpu python=3.11
# 激活虚拟环境
conda activate pytorch_gpu

cu118/torch-2.2.2%2Bcu118-cp311-cp311-win_amd64.whl
cu118：cuda版本11.8
torch-2.2.2：torch版本2.2.2
cp311：python版本3.11

在这里插入图片描述

进入下载torch所在目录安装
在这里插入图片描述

pip install .\torch-2.2.2+cu118-cp311-cp311-win_amd64.whl
pip install .\torchvision-0.17.2+cu118-cp311-cp311-win_amd64.whl

安装成功界面
在这里插入图片描述