首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

tensorflowpytorch临时指定CUDA版本,解决版本适配问题

经典的深度学习引擎,如tensorflow和pytorch,其自身版本CUDA版本有着严格的对应关系,一点点的不符都会使得程序无法运行。...,只靠一个版本CUDA来一招鲜吃遍天是不现实的,必然要安装多版本CUDA。...然而,指定CUDA版本也很繁琐,常规方法是开多个conda虚拟环境, 为每个环境指定一个CUDA版本,但不免繁琐。...其实,一个简单的方法就能够临时切换使用的CUDA版本,就是在执行代码前指定环境变量LD_LIBRARY_PATH为你想使用的特定版本CUDA的路径。...例如,为run.py指定使用CUDA=10.0: LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64 python run.py 其他版本,把cuda-10.0改成对应版本即可

2.4K10

Pytorch版本CUDA版本与显卡驱动版本的对应关系

参考链接:INSTALLING PREVIOUS VERSIONS OF PYTORCH 解决PyTorch与CUDA版本不匹配 1.CUDA驱动和CUDAToolkit对应版本 注:驱动是向下兼容的...,其决定了可安装的CUDA和CUDAToolkit的最高版本。...2.CUDA及其可用PyTorch对应版本(参考官网,欢迎评论区补充) 注:虽然有的卡CUDA版本可更新至新版本,且PyTorch也可对应更新至新版本。...3.安装指导 (1)指定安装PyTorch版本 当已知CUDA版本时,可根据表2直接查询到对应版本PyTorch,运行conda install pytorch=X.X.X -c pytorch即可安装指定版本...但不能保证PyTorch可正常使用,CUDAToolkit版本不适配显卡驱动,即可能导致CUDAToolkit版本高于CUDA驱动。

8.2K50

tensorflow各个版本CUDA以及Cudnn版本对应关系

(我采用的)即本地安装, 当我们选择离线安装,当我们选定相对应的版本之后,下载的时候发现这个地方的文件大小大概在2G左右,Linux系统下面我们选择runfile(local) 完整安装包从本地安装,或者是选择...CUDA 9.2、CUDA 10.0三个版本。...直接通过文件查看,这里分为Linux和windows两种情况 在windows平台下,可以直接进入CUDA的安装目录,比如我的是: C:\Program Files\NVIDIA GPU Computing...Toolkit\CUDA\v9.2 里面有一个version.txt的文本文件,直接打开即可,也可以使用命令,即 首先进入到安装目录,然后执行:type version.txt 即可查看 在Linux...7.5.0版本; (2)在Linux下当然也可以直接查看,但是通过命令更简单,进入到安装目录,执行如下命令: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR

3.8K20
领券