5

pytorch 配置详细过程 - E_sheep

1 year ago

source link: https://www.cnblogs.com/E-Sheep/p/17068878.html
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

neoserver,ios ssh client

torch

github 项目多
方便，api好调用

cpu版本

装torch

安装最新版本的就可以。

torchvision

要版本对应
算法：
torchvision版本号=
torch版本号第一个数字-1.torch版本号第二个数字+1.torch版本号第三个数字

所以我的就是：

pip install torchvision==0.14.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

成功效果：

GPU版本：

CUDA版本

先要确定CUDA:

nvidia-smi

这个是最高支持的版本：

然后我们去官网查看最新要求的版本：

可以看到是11.6

所以我们去网站上下载去11.6的
https://developer.nvidia.com/cuda-downloads

依次选择，然后下载就行

 nvcc -V

查看已安装的版本

https://developer.nvidia.com/cuda-toolkit-archive
下载历史版本

下载11.7版本

安装好之后，检查：

手动安装cuda的torch

http://download.pytorch.org/whl/torch/

下载对应版本的包。

torch version:

CPU的：

GPU的：

cu117就是cuda 11.7.x

python 版本：

cp38 就是 py 3.8.x

查看py版本：

我是3.9+

然后我是cuda11.7

windows平台

选定安装版本

开始安装：

成功

安装torchvision

http://download.pytorch.org/whl/torchvision/

成功：

测试GPU安装效果

import torch
print(torch.cuda.is_available())

返回True就是成功了

然后看一下 pip list

后面有这个加号就是可以的。

大功告成！！！！

__EOF__

本文作者： 小博的blog 本文链接： https://www.cnblogs.com/E-Sheep/p/17068878.html 关于博主： 评论和私信会在第一时间回复。或者直接私信我。 版权声明： 本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！ 声援博主： 如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。

Recommend

About Joyk

Aggregate valuable and interesting links.
Joyk means Joy of geeK