当前位置: 代码迷 >> 综合 >> RTX3090 windows cuda11.0 cudnn pytorch nightly试用版
  详细解决方案

RTX3090 windows cuda11.0 cudnn pytorch nightly试用版

热度:52   发布时间:2024-03-07 03:42:31.0

 

RTX3090到货了,花了一点时间重新装了cuda 和配了一下环境参考了这一篇文章:虽然标题写的是劝退文,但是他更新了说10.15号可以安装torch

https://blog.csdn.net/q1042960848/article/details/107856596

https://blog.csdn.net/lp1003/article/details/107240520/

1.下载cuda11.0

首先去NVIDA官网下载CUDA:

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

这里我下载的是现在最新的V11.0版本

-----

安装CUDA:

安装时选择:自定义安装-》只选择第一个-》一直到最后安装完成即可。主要是下面这一步的问题

这里可以参考:https://blog.csdn.net/shuiyixin/article/details/99935799

2.卸载cuda以前的版本

卸载cuda 我是在电脑里面的 控制面板--程序与安装里面--把与新版本重复的cuda都卸载了,还看了安装时间

然后再去环境路径中,把cuda10.0的路径删掉!这是时候一定要重启,这样才可以更新环境路径

2.下载cudnn

安装好后再在NVIDA官网上下载cuDNN,下载这个的时候需要注册。不需要翻墙也可以注册:

https://developer.nvidia.com/rdp/cudnn-download

这里我下载的是V8.0.1版本

一般来说下载的时候官网会根据做出的系统选项来推荐合适的CUDA以及cuDNN版本。

当然自己也可以根据本身电脑需求合理选择下载。

而cuDNN就是一些文件,将其中的文件复制到CUDA安装路径相应的目录下即可。不用担心会产生替换的问题。

3.卸载之前的pytorch

在conda uninstall pytorch 

4. 使用安装测试版本的pytroch 感谢知乎上的热心网友

pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu110/torch_nightly.html

5.然后在测试的时候有个小问题,大概是在import numpy的时候显示路径不对,大概要重新

conda uninstall numpy

conda install numpy 

就好啦!

import torch
print(torch.rand(3,3).cuda())

 

 

  相关解决方案