conda安装GPU版本的pytorch
安装cuda
从官网上下载或者从清华镜像下载都可以;
安装对应Ubuntu系统和nvidia驱动匹配的cuda系统:
安装cuda
conda install cudatoolkit == 10.1
安装Cudnn
安装cudnn:
conda install cudnn
激活cudnn:
打开.bashrc文件,在文件内部添加:
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"
再激活:
source ~/.bashrc
安装pytorch
一定要找到对应的版本进行安装,不能直接使用指令:
conda install torch
做法:
去torch的官网上找到和cuda版本对应的torch下载指令,进行下载;
如果没有的话可以去previous版本里面进行查找;
测试:
如果需要使用GPU的需要测试能否在GPU上使用:
python
import torch
torch.cuda.is_available()
如果结果是True就表示可以在GPU上使用。
切记,一定要按照上述的方法,对应好所有软件的版本号,否则torch.cuda.is_available()都有False的风险。
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhfijjbf
系列文章
更多
同类精品
更多
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01 -
怎样阻止微信小程序自动打开
PHP中文网 06-13