0
点赞
收藏
分享

微信扫一扫

高质量床上用品类网站带手机端的pbootcms模板

开源分享 2023-10-17 阅读 12

【超级简单】3步 安装conda + pytorch gpu版本

下载anaconda,能让你高效管理不同版本运行环境。
conda + pytorch gpu
适用于linux 和window版本,
建议前提确保自己的显卡能用哈。

以下是在window10的示范。

1. 创建虚拟环境

conda create -n your_env_name python=3.X
conda avtivate your_env_name

至于python 3.X和cuda对应 ?

参考Tensorflow 的资料:
https://www.tensorflow.org/install/source_windows?hl=zh-cn

2. 下载cuda 驱动,cuda toolkit

https://developer.nvidia.com/cuda-toolkit-archive

下载成功的话可以 在cmd 输入 nvcc -V, 显示版本号。

显示 not found 则说明没转成功。

3. 虚拟环境中输入以下指令

conda install pytorch torchvision torchaudio cudatoolkit=X.X  -c pytorch

X.X替换成你的版本号

至于 怎么查看自己电脑对应的cuda版本号 ?

:鼠标右键——>NVDIA控制面板——>帮助——>系统信息——>组件,product name 那一列就是nvdia对应版本。

有疑惑可以再看看这篇文章写的。

笔墨多一点。
具体可参考:https://blog.csdn.net/qq_36851515/article/details/107608967

举报

相关推荐

0 条评论